AMD GPU

#4
by Regrant-Dev - opened

I have rocm and pytorch all setup on a linux machine. I can't seem to get this to run on my amd gpu (vega56) though. Is there any way to do this currently?

Databricks org

Just about all of these models you'll encounter need CUDA, hence NVIDIA GPUs. There are some ports of pytorch et al to AMD (ROCm et al) that you can try, but I think other machinery still relies on CUDA, like deepspeed, accelerate, etc.

Thanks for the quick reply! Looks like i will be investing in an NVIDIA gpu then.

Regrant-Dev changed discussion status to closed
Databricks org

Pytrch models will work on CPU too yes, albeit slowly. Are you getting it to work on built in Intel graphics? Can work sometimes on other GPUs with things like ROCm but my experience is that these models often have some CUDA specific element that is needed to run fast

Pytrch models will work on CPU too yes, albeit slowly. Are you getting it to work on built in Intel graphics? Can work sometimes on other GPUs with things like ROCm but my experience is that these models often have some CUDA specific element that is needed to run fast

Yes, works on both GPU and non GPU CPU from Intel
3b responds in up to 1 minute, 7b - 3-5 minutes, 12b - 15-75 minutes depending from the prompt

Sign up or log in to comment