CMake Error at libtorchsharecmakeCaffe2publiccudacmake55 message PyTorch requires CUDA 110 or aboveCall Stack most recent call first libtorchsharecmakeCaffe2Caffe2Configcmake88 include libtorchshar
这个错误提示表明你的PyTorch版本需要CUDA 11.0或更高版本。要解决这个问题,你可以采取以下步骤:
-
确保你已经安装了CUDA 11.0或更高版本。你可以通过在终端中运行
nvcc --version来检查CUDA的安装版本。 -
如果你尚未安装CUDA 11.0或更高版本,请访问NVIDIA的官方网站(https://developer.nvidia.com/cuda-toolkit-archive)下载并安装适用于你的操作系统的相应版本。
-
如果你已经安装了CUDA 11.0或更高版本,但是仍然收到这个错误消息,请确保你的系统环境变量中设置了正确的CUDA路径。你可以通过在终端中运行
echo $PATH和echo $LD_LIBRARY_PATH来检查环境变量的设置情况。 -
如果你的环境变量没有设置正确的CUDA路径,可以通过编辑
~/.bashrc或~/.bash_profile文件,并添加以下行来设置CUDA路径:
export PATH=/usr/local/cuda-X.X/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-X.X/lib64:$LD_LIBRARY_PATH
请确保将X.X替换为你安装的CUDA版本号。
-
保存并关闭文件后,执行
source ~/.bashrc或source ~/.bash_profile来加载新的环境变量。 -
在重新编译你的项目之前,确保在CMakeLists.txt文件中正确设置了PyTorch的路径。你可以通过检查CMakeLists.txt文件中的
find_package(Torch)语句来确认路径是否正确。 -
清除你的构建目录并重新运行CMake命令以重新生成项目文件。
通过执行上述步骤,你应该能够解决这个问题并成功编译你的项目
原文地址: https://www.cveoy.top/t/topic/ibbN 著作权归作者所有。请勿转载和采集!