How to use pytorch3d with an amd gpu? #750
Unanswered
joaodalino
asked this question in
Q&A
Replies: 1 comment
-
|
The compiled parts of PyTorch3D don't include ROCm support. You should be able to build a CPU-only version of PyTorch3D, and you should have no problem installing a prebuilt PyTorch3D if you are on Linux. But any of the compiled parts of PyTorch3D will only work with CPU tensors, and I would not be surprised if they crash if given GPU tensors. Can you say more about your install and what went wrong? |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hey, i'm newbie in 3d meshing. I was trying to install pytorch3d without having a nvidia gpu, but when i proceed to the installation, my pc just crash. It's possible to use it with an amd gpu?
Beta Was this translation helpful? Give feedback.
All reactions