Windows下Pytorch3安装记录

1 安装重点

要求cudatoolkit、cuda、pytorch、NVIDIA CUB 版本的严格对应

2 安装环境与版本

python=3.7.13
pytorch=1.12.1+cu116
pytorch3d=0.7.1（官网下载地址，版本对应官网有写）
cudatoolkit=11.7
NVDIA CUB=1.16.0（官网下载地址，版本对应如下图）

3 安装过程

1、前置工作
①下载pytorch3d的安装包(zip)并解压
②cuda版本在11.6以上的替换CUB包：下载CUB的包(zip)并解压，在cuda路径下（一般在C盘，我的是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\include\cub）替换掉原来的CUB包
或
cuda版本在11.6以下的，直接添加cub路径到控制面板-系统变量的CUB_HOME=cub包所在的路径
③修改pytorch3d安装包路径下的 ’setup.py’ 文件，注释掉 extra_compile_args = {"cxx": ["-std=c++14"]}
④检查python和pytorch安装情况
激活环境后输入import torch和print(torch.__version__,torch.cuda.is_available())<