pytorch3d 配置

尝试windows下,linux下pytorch3d的安装,各种不成功,每次我都不是幸运的那个一次成功,这样有幸记录下自己的坎坷,希望大家可以节省时间在配置上。

windows

一、pip install pytorch3d 可以成功,但是是0.01版本,太老了,库比较少,所以基本淘汰了。

二、在github下载的v0.2.0版本
python3.6.8 + cuda10.1+ cudnn7 + pytorch1.7.1+cu101 (其他版本应该影响不大)

  1. 下载pytorch3d 到本地,下载release版本 https://github.com/facebookresearch/pytorch3d/archive/v0.2.0.zip

  2. 以管理员方式打开x64 Native Tools Command Prompt for VS 2019

  3. 进入pytorch3d目录
    cd pytorch3d-0.2.0
    python setup.py install

出现以下语句即为安装成功,然而0.2.0包还不完善
Using d:\programs\python3.6.8\lib\site-packages
Finished processing dependencies for pytorch3d==0.2.0

————————————————————————————————————————
也许你会遇到下面的问题
File "D:\Programs\python3.6.8\lib\site-packages\torch\utils\cpp_extension.py", line 1529, in _run_ninja_build
raise RuntimeError(message)
RuntimeError: Error compiling objects for extension

按照这里面的 https://github.com/facebookresearch/pytorch3d/issues/323

同样方式0.4.0暂时没有成功,待更新

Linux

一、作者已经编译了python3.6,3.8 和 pytorch171,170版本的库,你可以修改版本试试看看存在不存在
https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py36_cu101_pyt170/download.html
pip install ./pytorch3d-0.4.0-cp36-cp36-linux_x86_64.whl

二、自己在python3.6,+pytorch1.5.0+cu101上没有自己编译成功,如果有编译成功的同胞可以分享一下。

引用:
https://www.icode9.com/content-3-1020527.html
https://github.com/facebookresearch/pytorch3d/issues/717

你可能感兴趣的:(pytorch3d 配置)