win10系统下安装PyTorch3d

最近在复现3d人体重建和人脸重建相关demo算法，发现都需要安装PyTorch3d，然而Linux环境借助Conda可以一键安装，对于Win10环境来说真实太痛苦了，博主折腾了几天，各种重建环境，现在已成功安装PyTorch3d。

一、环境版本一致性要求

环境版本最好保持一致性，省的来回更改环境，以至于最后把系统环境改的面目全非。

NVIDIA CUDA，NVIDIA CUB，Visual Studio ，Pytorch，Pytorch3d必须需要版本一致

首先官网对于Pytorch3d在windows安装说明

本人显卡为3090，NVIDIA CUDA是cuda11.3，Conda中pytorch=1.8.0，cuda=11.1，Visual Studio =2019，Pytorch3d==0.6.0；NVIDIA CUB=1.11.0

其中NVIDIA CUDA11.3支持Visual Studio2017和2019，安装2022会报版本不一致,所以最好还是老老实实安装2019，其中NVIDIA CUB官网推荐1.10.0，这里需要更改为1.11.0不然会报版本不一致。

 

注意：

1. 把setup.py 中[‘ninja’, ‘-v’]改为[‘ninja’, ‘–version’]是错误的
2. 不用在系统环境变量中配置Visual Studio2019 的cl.exe，直接通过下图所示启动命令窗口，然后输入set DISTUTILS_USE_SDK=1，同时进入pytorch3d文件夹：

 二、提前修改相关文件

1.pytoch3d中源码更改

setup.py 中43行：

 extra_compile_args = {"cxx": ["-std=c++14"]}注释掉改为extra_compile_args = {"cxx": []}

setup.py 中60行：-std=c++14注释掉

 iou_box3d.cu修改注释掉>#include "utils/pytorch3d_cutils.h"

 iou_utils.cuh和iou_utils.h搜索not，把not改为！，最新版本PyTorch3d已经解决这个问题

 更改后可以解决如下错误：

2.pytorch源码中更改 

需要找到cpp_extension.py进行修改，本人如下：

C:\Users\dell\miniconda3\envs\icon\lib\site-packages\torch\utils\cpp_extension.py

三、命令窗口安装编译

在命令窗口输入 python setup.py install 进行安装编译

四、参考

1. https://blog.csdn.net/zzqkz20121221/article/details/121157357
2. https://github.com/facebookresearch/pytorch3d/issues/713
3. https://zhuanlan.zhihu.com/p/371279126
4. https://mp.csdn.net/mp_blog/creation/editor/125011964
5. https://zhuanlan.zhihu.com/p/371279126
6. https://blog.csdn.net/qq_40714949/article/details/116571010
7. https://blog.csdn.net/weixin_41793473/article/details/118669832