Windows10+Conda+CUDA10.1编译Pytorch3D

pytorch3D 0.6.0终于在windows上编译通过了，写下博客记录一下，免得后面又忘记！

我的参考：installation - Installing PyTorch3D fails with anaconda and pip on Windows 10 - Stack Overflow

第一步：按照官网readme进行配置

conda create -n pytorch3d python=3.8
conda activate pytorch3d
conda install -c pytorch pytorch=1.7.1 torchvision cudatoolkit=10.2
conda install -c fvcore -c iopath -c conda-forge fvcore iopath

第二步：https://github.com/NVIDIA/cub/releases上下载CUB_LIB并配置环境变量，直接下载解压后将路径配置在环境中，我的

第三步：设置cl.exe的环境变量，参考：

D:\Program Files(x86)\MicrosoftVisualStudio\2019\Professional\VC\Tools\MSVC\14.28.29910\bin

第四步：打开本地的“x64 Native Tools Command Prompt for VS 2019”命令窗，在解压的pytorch3d 文件夹下使用conda里面的环境执行：

D:\Program Files (x86)\Microsoft Visual Studio\2019\Professional>cdD:\python_lib\pytorch3d-main
D:\python_lib\pytorch3d-main>conda activate pytorch-gpu-py3-8
(pytorch-gpu-py3-8) D:\python_lib\pytorch3d-main>python setup.py install

根据提示需要在命令框中输入一次输入以下宏设置：

SET DISTUTILS_USE_SDK=1
SET FORCE_CUDA=1
SET PYTORCH3D_NO_NINJA=1

根据提示需要修改C:\Anaconda\envs\pytorch-gpu-py3-8\Lib\site-packages\torch\include \torch\csrc\jit\api中的module.h文件

line 483：const CONSTEXPR_EXCEPT_WIN_CUDA bool all_slots = false; 
          改为：static const CONSTEXPR_EXCEPT_WIN_CUDA bool all_slots = false; 
line 496，510， 523同样的道理

最后在执行一次：

python setup.py install

不出错的话就会在conda环境中自动生成pytorch3d-0.6.0-py3.8-win-amd64.egg文件夹。

安装到此结束！！！！