安装Visual Studio
安装的时候一定要选择language C++。
安装成功后,将cl.exe的路径添加到环境变量Path中。
一般cl.exe的地址是在Microsoft Visual Studio***\VC\bin\amd64\里面。
安装依赖项
conda create -n pytorch3d python=3.10
conda activate pytorch3d
conda install pytorch=1.13.0 torchvision pytorch-cuda=11.6 -c pytorch -c nvidia
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
如果CUDA版本低于11.7,需要安装CUB
conda install -c bottler nvidiacub
安装pytorch3d
下载pytorch3d源码
git clone https://github.com/facebookresearch/pytorch3d.git
以管理员身份打开“x64 Native Tools Command Prompt...”
在conda环境下,进入pytorch3d目录执行编译
conda activate pytorch3d
cd pytorch3d
python setup.py install
检测是否安装成功:
import pytorch3d
print(pytorch3d.__version__)