0.缘由
- 重新安装pytorch3d的时候,踩了超多坑,特此总结一下自己的安装,希望能对大家有帮助。
- 弱弱地吐槽一下,这段时间用Windows+子系统,比用Ubuntu舒服多了,毕竟没事还能打打游戏哈哈哈哈。
1.安装pytorch
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
- 这里一定要注意,必须用pip安装!!
- 用conda安装虽然能成功,但在装pytorch3d的时候会有一堆奇奇怪怪的错误TOT。
2.安装pytorch3d
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
- 再在Github下载pytorch3d,并解压到一个单独的文件夹里,我命名为pytorch3d;(别问为什么不用wget,个人喜好O_o)
- cd到pytorch3d文件夹中,用pip或者setup.py安装(我这里用的是pip)
pip install -e .
或者
python3 setup.py install
注意事项
- 安装的时候可能会有“subprocess.CalledProcessError: Command ‘[‘ninja‘, ‘-v‘]‘ returned non-zero exit status 1”的错误,这里参考这篇blog,把setup.py里的
cmdclass={'build_ext': BuildExtension}
改成
cmdclass={'build_ext': BuildExtension.with_options(use_ninja=False)}
- 安装好后的pytorch3d文件夹,一定一定不要删掉!!!!