当前位置: 首页 > 工具软件 > PyTorch3D > 使用案例 >

Linux子系统下的pytorch3d安装记录(miniconda安装,pytorch1.7.1版本)

邹嘉致
2023-12-01


0.缘由

  • 重新安装pytorch3d的时候,踩了超多坑,特此总结一下自己的安装,希望能对大家有帮助。
  • 弱弱地吐槽一下,这段时间用Windows+子系统,比用Ubuntu舒服多了,毕竟没事还能打打游戏哈哈哈哈。

1.安装pytorch

  • Linux子系统下安装
  • 安装、激活、删除conda环境可以参考这篇blog
  • 在pytorch官网中的Previous PyTorch Versions页面能找到安装指令,因为我现在用的是3090,所以用下面这个指令安装:
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

  • 这里选择本地安装,参考的是这篇blog
  • 先安装核心库:
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文件夹,一定一定不要删掉!!!!
 类似资料: