这个错误提示表明你的PyTorch版本需要CUDA 11.0或更高版本。要解决这个问题,你可以采取以下步骤:

  1. 确保你已经安装了CUDA 11.0或更高版本。你可以通过在终端中运行nvcc --version来检查CUDA的安装版本。

  2. 如果你尚未安装CUDA 11.0或更高版本,请访问NVIDIA的官方网站(https://developer.nvidia.com/cuda-toolkit-archive)下载并安装适用于你的操作系统的相应版本。

  3. 如果你已经安装了CUDA 11.0或更高版本,但是仍然收到这个错误消息,请确保你的系统环境变量中设置了正确的CUDA路径。你可以通过在终端中运行echo $PATHecho $LD_LIBRARY_PATH来检查环境变量的设置情况。

  4. 如果你的环境变量没有设置正确的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版本号。

  1. 保存并关闭文件后,执行source ~/.bashrcsource ~/.bash_profile来加载新的环境变量。

  2. 在重新编译你的项目之前,确保在CMakeLists.txt文件中正确设置了PyTorch的路径。你可以通过检查CMakeLists.txt文件中的find_package(Torch)语句来确认路径是否正确。

  3. 清除你的构建目录并重新运行CMake命令以重新生成项目文件。

通过执行上述步骤,你应该能够解决这个问题并成功编译你的项目

CMake Error at libtorchsharecmakeCaffe2publiccudacmake55 message PyTorch requires CUDA 110 or aboveCall Stack most recent call first libtorchsharecmakeCaffe2Caffe2Configcmake88 include libtorchshar

原文地址: https://www.cveoy.top/t/topic/ibbN 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录