search

facebookresearch / pytorch3d

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
https://pytorch3d.org/
Other
8.7k stars 1.3k forks source link

Does PyTorch3D support AMD Radeon W7900? (gfx1100) #1845

Closed yttbgf closed 1 month ago

yttbgf commented 2 months ago

Does PyTorch3D support AMD Radeon W7900? (gfx1100) I see the code include at::cuda::CUDAGuard (*.cu) and commands.h include #include <cub/cub.cuh> I want to know whether is can port to AMD ROCm with HIP? Does it use inline assemble (asm） or OPENGL like nvdiffrast-main

bottler commented 2 months ago

Inline assembly is not used. The compiled parts of pytorch3d do not use opengl, but there is a mesh renderer in python which uses opengl - but you can use the rest of pytorch3d without being affected by it. AMD is not supported but you might be able to get it to work.