Ubuntu下nvcc CUDA版本显示异常及PyTorch版本不匹配问题

1. 确认已安装的CUDA版本 #

先查看系统里已安装的CUDA Toolkit目录：

ls /usr/local/ | grep cuda

你应该能看到cuda-10.1、cuda-11.7、cuda-12.1这类目录，其中cuda是指向默认版本的软链接。

2. 切换默认CUDA版本到11.7 #

方式一：修改软链接 #

先删掉当前的默认软链接：

sudo rm /usr/local/cuda

再创建指向11.7版本的新软链接：

sudo ln -s /usr/local/cuda-11.7 /usr/local/cuda

方式二：调整环境变量 #

编辑你的shell配置文件（比如~/.bashrc或~/.zshrc），找到原来的CUDA相关环境变量行，把其中的cuda-10.1替换成cuda-11.7；如果没有相关配置，就添加以下内容：

export PATH=/usr/local/cuda-11.7/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH

保存后刷新环境变量：

source ~/.bashrc

3. 验证切换结果 #

重启终端后执行以下命令，确认nvcc版本已切换到11.7：

nvcc -V

也可以检查环境变量是否生效：

echo $PATH | grep cuda
echo $LD_LIBRARY_PATH | grep cuda

4. 处理apt安装冲突 #

你执行sudo apt-get install cuda时显示最新版是12.1，是因为apt源默认拉取最新CUDA包。如果要通过apt安装特定版本的CUDA，需要指定版本号：

sudo apt-get install cuda-11-7

不过你已经手动下载安装了11.7版本，这一步可以跳过，只要确保软链接和环境变量配置正确即可。

5. 重新安装torch-cluster #

确认nvcc版本为11.7后，重新安装torch-cluster，建议匹配你的PyTorch版本，比如用pip安装：

pip install torch-cluster -f https://data.pyg.org/whl/torch-1.13.0+cu117.html

（注意替换成你实际使用的PyTorch版本对应的CUDA11.7安装源，环境变量配置正确的情况下，pip会自动匹配CUDA版本）

内容的提问来源于stack exchange，提问作者MikeB2019x