解决的问题:服务器上跑深度学习代码,通常都需要用到Cuda。有时候跑的不同程序要求的配置Cuda版本可能也不同,会需要不同Cuda版本的替换。
Linux安装Cuda
CUDA官网,下载安装,网址为:https://developer.nvidia.com/cuda-11.1.0-download-archive。
根据自己的本地环境配置就行选择。
下面会显示两行下载和安装的指令,按顺序输入。
wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run
sudo sh cuda_11.1.0_455.23.05_linux.run
之后耐心等待一会,用上下键选中“Continue”回车。
然后用户同意部分,输入“accept”。
如果已经装过cuda,只是装多一个版本,可以把第一个Driver给取消掉,然后选中“install”安装。
之后一直默认就好了。
多个Cuda版本切换
很简单,就几行代码的事情。
先查看一下有多少个cuda版本,我们需要切换的是11.1.
查看安装的所有cuda版本。
ls /usr/local | grep cuda
打开bashrc文件。
vim ~/.bashrc
在bashrc文件最后添加如下,如果是切换成其它版本把,cuda后面的版本号改了就行了。
export PATH="/usr/local/cuda-11.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH"
esc后,输入":wq"保存后退出。
更新环境变量。
source ~/.bashrc
查看当前cuda版本,看是否切换:
nvcc -V
看到已经是cuda11.1,cuda版本切换完成。