Win10下pytorch3D安装方法，本人亲测可用

本人的电脑配置如下：

GPU：GTX 1050Ti

CUDA：11.1

python：3.8

pytorch：1.9.0

pytorch3d：0.7.1

NVIDIA CUB：cub-1.9.10-1

目录

1.版本确认

2.下载cub

3.下载pytorch3d

 4.下载VSC2019

 5.编译pytorch3d源文件

---

1.版本确认

        这个步骤非常非常重要，因为不同的pytorch3d版本适配于不同的torch版本，你可以进到facebook/pytorch3d的官网查看不同的适配版本（这是0.7.1的），其余的你们可以自己查看https://github.com/facebookresearch/pytorch3d/releases/tag/v0.7.1

 此外，你要保证你的CUDA环境变量要对，比如我的CUDA是11.1，我的cuda toolkit也是11.1的，并把它添加到你的用户的环境变量里，此外还有你的conda也要添加到用户环境变量里面，因为后面我们会用到一个叫做Microsoft Visual Studio2019的工具，在使用时需要你的conda命令进行激活。如果你们有和我一样的版本，cuda11.1，你可以自己新建一个环境，然后重新装一个torch：

pip --default-timeout=1000 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

其它依赖：

conda install -c fvcore -c iopath -c conda-forge fvcore iopath

这样我们就有了一个1.9.0的torch。

 你们也可以自己输入这段代码验证：

import torch
print(torch.cuda.is_available())
print(torch.__version__)

2.下载cub

        CUDA和NVIDIA CUB版本对应关系如下：GitHub - NVIDIA/cub: Cooperative primitives for CUDA C++.

下载好后，将其加入自己的环境变量：

3.下载pytorch3d

        根据你自己的需要下载对应版本的pytorch3d压缩包

下载好后，解压放在文件夹，我是放在了e盘 

 4.下载VSC2019

        因为下载好的源文件需要用c++进行编译VS2019是最好的编译的版本，我们可以去官网下载，安装时，我们选择默认的C++的安装选项

 5.编译pytorch3d源文件

        这个部分是最最关键的，我个人是这么觉得的！

第一步：找到x64 tool vs2019 prompt：

 注：千万要用管理员的方式运行，不然编译的时候头文件调用会出问题！！

 第二步：

cd 你源文件文件夹的位置

 此外，你需要进入conda环境，就像我前面有个（pytorch3d）一样！！

这就提到了我们前面说的为什么要把conda添加到环境变量里面，直接：

conda activate (你创建的环境，我这里是pytorch3d)

接着输入这两段（为了防止编译超时）：

set DISTUTILS_USE_SDK=1
set PYTORCH3D_NO_NINJA=1

最后运行：

python setup.py install

x64 native tool vs2019 prompt会开始编译，差不多10分钟就可以编译好：

 到这里，pytorch3d的安装就结束啦，有其他问题的小伙伴可以评论区问我哦！