pytorch3d icon indicating copy to clipboard operation
pytorch3d copied to clipboard

How to install Pytorch3d with Cuda11.6 and pytorch1.12

Open MrTornado24 opened this issue 2 years ago • 1 comments

🐛 Bugs / Unexpected behaviors

I tried to install pytorch3d with cuda 11.6 and pytorch1.12. The python version is 3.9. I used local git method pip install -e ., however, it failed with some compiling errors. I wonder if pytorch3d supports cuda 11.6 and pytorch1.12.

MrTornado24 avatar Aug 07 '22 02:08 MrTornado24

I had the same issue. A temporary fix is following the README instructions and then downgrading PyTorch: pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 -f https://download.pytorch.org/whl/torch_stable.html.

I had another issue requiring another version downgrade, too: python3 -m pip install setuptools==59.5.0

Test it out with the following code as you go: torch.zeros(1).cuda()

Edit: I admittedly discarded this setup and started anew. I've found that CUDA 11.3 and PyTorch 1.11 are stable if you follow the instructions and substitute in the respective versions. PyTorch 1.12 seems problematic, can't speak for the newer CUDA versions. Did you get anywhere with this?

domattioli avatar Aug 09 '22 20:08 domattioli