今天在编译安装pytorch3D的过程中遇到以下报错
3 errors detected in the compilation of "C:/Projects/pytorch3d/pytorch3d/csrc/pulsar/cuda/renderer.backward.gpu.cu".
error: command 'C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin\nvcc.exe' failed with exit code 2
PS C:\Projects\pytorch3d>
解决方案:
1. 安装CUDA11.7版本,自己原来的CUDA版本可以不用删除(自己可以谷歌如何同时保留两个CUDA版本并自由切换),安装完成后使用:
nvcc --version
查看是否成功安装;
2. 在CUDA11.7/include/cub文件夹下,打开 version.cuh, 添加:#define THRUST_IGNORE_CUB_VERSION_CHECK 在#pragma once下:
#pragma once
#define THRUST_IGNORE_CUB_VERSION_CHECK
3.编译安装PyTorch3D