-
I have an AMD GPU. I have installed the PyTorch ROCm version and PYG CPU version. |
Beta Was this translation helpful? Give feedback.
Replies: 4 comments 32 replies
-
We added ROCm support to
Let me know if this works. |
Beta Was this translation helpful? Give feedback.
-
We don't control these packages and how the wheels are prepared and delivered -- there are no ROCm versions of these wheels that can be pip installed. You must build all of the prereqs from source. Be careful with using pip in case it installs a CUDA version of the wheel instead. The missing thrust header error needs some triage. The file csrc/hip/segment_csr_hip.hip appears to have been hipified, but didn't convert the thrust include to a hipthrust include. I'll need to reproduce these steps to confirm what's going wrong. |
Beta Was this translation helpful? Give feedback.
-
inside docker container based on image rocm/pytorch:latest
This was ubuntu 20, rocm 5.4.0. |
Beta Was this translation helpful? Give feedback.
-
@rusty1s These are the wheels I built for PyTorch 1.13 from py3.7 to py3.10. |
Beta Was this translation helpful? Give feedback.
inside docker container based on image rocm/pytorch:latest
This was ubuntu 20, rocm 5.4.0.