在Win11下安装pytorch3d

本文主要是介绍在Win11下安装pytorch3d，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

安装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__)

这篇关于在Win11下安装pytorch3d的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---