Windows安装Pytorch3d

Windows安装Pytorch3d

1.前提：

1. 安装Visual Studio 2019 【我记得必须是2017-2019之间的版本，我一开始用的是2022的版本就安装不了】网址
2. 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版本，下载网址：

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

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

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版本

下载此压缩文件并解压到自己想安装的位置，我解压的位置是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未报错

参考资料

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