Linux子系统下的pytorch3d安装记录

---

0.缘由

• 重新安装pytorch3d的时候，踩了超多坑，特此总结一下自己的安装，希望能对大家有帮助。

---

1.安装pytorch

• Linux子系统下安装
• 安装、激活、删除conda环境可以参考这篇blog；
• 在pytorch官网中的Previous PyTorch Versions页面能找到安装指令，我用下面这个指令安装：

pip3 install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu111

• 这里一定要注意，必须用pip安装！！
• 用conda安装虽然能成功，但在装pytorch3d的时候会有一堆奇奇怪怪的错误.
• 另外，也一定要安装对应版本的cudatoolkit：

conda install cudatoolkit=11.1

---

2.安装pytorch3d

• 这里选择本地安装，参考的是这篇blog；
• 先安装核心库：

conda install -c fvcore -c iopath -c conda-forge fvcore iopath

• 再在Github下载对应版本的pytorch3d，我用的0.6.2版本，并解压到一个单独的文件夹里，我命名为pytorch3d；
• 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文件夹，一定一定不要删掉！！！！