Windows安装Pytorch3d-Toy模板网

这篇具有很好参考价值的文章主要介绍了Windows安装Pytorch3d。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

Windows安装Pytorch3d

1.前提：

安装Visual Studio 2019 【我记得必须是2017-2019之间的版本，我一开始用的是2022的版本就安装不了】网址
pytorch和pytorch3d、cuda和NVIDIA CUB版本需要相互对应

pytorch和pytorch3d版本对应关系如下：https://github.com/facebookresearch/pytorch3d/releases

cuda和NVIDIA CUB版本对应关系如下：https://github.com/NVIDIA/cub/releases?page=1

本人配置如下：

GPU：1050Ti

CUDA：11.1

python：3.8

pytorch：1.8.0

pytorch3d：0.7.0

NVIDIA CUB：cub-1.9.10-1

2.安装

2.1 创建虚拟环境

首先打开cmd命令窗口，创建并激活虚拟环境

conda create -n pytorch3d python=3.8
conda activate pytorch3d

2.2 安装pytorch

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

2.3 安装其他依赖

conda install -c fvcore -c iopath -c conda-forge fvcore iopath
conda install jupyter
pip install scikit-image matplotlib imageio plotly opencv-python
pip install black usort flake8 flake8-bugbear flake8-comprehensions

2.4 下载cub

cuda和NVIDIA CUB版本对应关系如下（不全，完整查看网址）：

CUB Release	Included In
2.0.1	CUDA Toolkit 12.0
2.0.0	TBD
1.17.2	TBD
1.17.1	TBD
1.17.0	TBD
1.16.0	TBD
1.15.0	NVIDIA HPC SDK 22.1 & CUDA Toolkit 11.6
1.14.0	NVIDIA HPC SDK 21.9
1.13.1	CUDA Toolkit 11.5
1.13.0	NVIDIA HPC SDK 21.7
1.12.1	CUDA Toolkit 11.4
1.12.0	NVIDIA HPC SDK 21.3
1.11.0	CUDA Toolkit 11.3
1.10.0	NVIDIA HPC SDK 20.9 & CUDA Toolkit 11.2
1.9.10-1	NVIDIA HPC SDK 20.7 & CUDA Toolkit 11.1
1.9.10	NVIDIA HPC SDK 20.5
1.9.9	CUDA Toolkit 11.0
1.9.8-1	NVIDIA HPC SDK 20.3
1.9.8	CUDA Toolkit 11.0 Early Access
1.9.8	CUDA 11.0 Early Access

根据自己的CUDA版本选择对应的cub版本，下载网址：

Windows安装Pytorch3d

下载此压缩文件并解压到自己想安装的位置

而后添加环境变量，变量值为解压缩的位置：

Windows安装Pytorch3d

2.5 下载pytorch3d

pytorch和pytorch3d版本对应关系如下：https://github.com/facebookresearch/pytorch3d/releases

这里的对应版本就看Version下第一行There are builds for PyTorch 1.12.0, but no longer 1.7.x.，可以看出最高支持1.12.0，且不再支持1.7.x的版本。

根据自己的pytorch版本选择对应的pytorch3d版本

Windows安装Pytorch3d

下载此压缩文件并解压到自己想安装的位置，我解压的位置是E:\pytorch3d-0.7.0（按照我之前看的博文最好应该解压到虚拟环境的Lib\site-packages目录下）

2.6 安装MinGW

其他教程中有包含这一步，我的电脑很久前就装过MinGW，所以此处省略。

2.7 安装pytorch3d

2.7.1 修改setup.py

打开pytorch3D文件夹，找到setup.py文件并打开，将extra_compile_args = {"cxx": ["-std=c++14"]} 修改为: extra_compile_args = {"cxx": []}

2.7.2 安装pytorch3d

使用管理员身份打开x64 Native Tools Command Prompt for VS 2019，进入pytorch3d文件夹的路径并激活虚拟环境。

cd E:\pytorch3d-0.7.0
E:
set DISTUTILS_USE_SDK=1
set PYTORCH3D_NO_NINJA=1
python setup.py install

等待代码执行完成即可。

2.8 验证

导入pytorch3d未报错
Windows安装Pytorch3d 文章来源地址https://www.toymoban.com/news/detail-494316.html

参考资料

pytorch3D Windows下安装经验总结
Windows下Pytorch3d的安装方法

到了这里，关于Windows安装Pytorch3d的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！

Windows安装Pytorch3d