【Cuda】Linux 升级 Cuda

1. 根据 Pytorch 版本选择相应的 Cuda 版本

https://pytorch.org/get-started/locally/#linux-prerequisites-2

如果原先是10.2版本，升级到11.x版本的时候，建议升级到11.3，避免与Pytorch不兼容的情况

2. 安装相应的 Cuda 版本

https://developer.nvidia.com/cuda-11.3.0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=runfile_local

uname -v 查看当前 Linux 系统的版本

3. 取消 Driver 的安装选项

continue -> accept -> 取消 Driver 选项

4. Cuda 环境配置

vim ~/.bashrc
#按o进入编辑模式，并在最后一行添加如下代码
export PATH=$PATH:/usr/local/cuda-11.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.0/lib64
#esc后输入wq退出保存

5. nvcc -V 配置

sudo vi /usr/bin/nvcc

替换 exec 之后的路径为 cuda 的环境配置路径

参考资料

https://blog.csdn.net/DK_csdn1/article/details/121288000