尝试windows下,linux下pytorch3d的安装,各种不成功,每次我都不是幸运的那个一次成功,这样有幸记录下自己的坎坷,希望大家可以节省时间在配置上。
windows
一、pip install pytorch3d 可以成功,但是是0.01版本,太老了,库比较少,所以基本淘汰了。
二、在github下载的v0.2.0版本
python3.6.8 + cuda10.1+ cudnn7 + pytorch1.7.1+cu101 (其他版本应该影响不大)
-
下载pytorch3d 到本地,下载release版本 https://github.com/facebookresearch/pytorch3d/archive/v0.2.0.zip
-
以管理员方式打开x64 Native Tools Command Prompt for VS 2019
-
进入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
网友评论