pytorch3d
pytorch3d copied to clipboard
How to install Pytorch3d with Cuda11.6 and pytorch1.12
🐛 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.
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?