I am trying to use PyTorch3D on a singularity image. I have successfully installed it with torch 2.0.0 and CUDA 11.7 from wheel (version_str = 'py38_cu117_pyt200'), but I would like to use it with torch 1.8.0 and CUDA 11.1, because of some compatibility with a C++ and CUDA extension. However, when I try to install it with these versions (either 'py38_cu111_pyt180a0' or 'py38_cu111_pyt180'), it is not able to find the wheel. I have also tried from source but does not seem to work. I would appreciate any suggestions on how to solve this.
❓ Questions on how to use PyTorch3D
Hi,
I am trying to use PyTorch3D on a singularity image. I have successfully installed it with torch 2.0.0 and CUDA 11.7 from wheel (version_str = 'py38_cu117_pyt200'), but I would like to use it with torch 1.8.0 and CUDA 11.1, because of some compatibility with a C++ and CUDA extension. However, when I try to install it with these versions (either 'py38_cu111_pyt180a0' or 'py38_cu111_pyt180'), it is not able to find the wheel. I have also tried from source but does not seem to work. I would appreciate any suggestions on how to solve this.
Thanks!