手动安装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

你可能感兴趣的:(pytorch,人工智能,python)