当前位置:   article > 正文

手动安装pytorch3D_pytorch3d安装

pytorch3d安装

一、安装两个核心库(手动安装)
 

iopath

git clone https://github.com/facebookresearch/iopath
​​pip install -e iopath


fvcore

git clone https://github.com/facebookresearch/fvcore
pip install -e fvcore


 


二、安装CUB libaray


命令如下:

conda install -c bottler nvidiacub
 

 

三、安装Pytorch3D(手动Build)


直接选择git clone到本地

git clone https://github.com/facebookresearch/pytorch3d.git
通过pip安装

cd pytorch3d
pip install -e .

通过setup.py安装

cd pytorch3d
python3 setup.py install
安装需要一点时间,如果中途不报错应该没有太大问题,如果build出现问题可以尝试更改gcc版本。

另外,可以使用官网的test命令测试build成功与否

cd pytorch3d/tests
python3 -m unittest discover -p *.py

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/盐析白兔/article/detail/324208
推荐阅读
  

闽ICP备14008679号