由于最近项目要求的cuda版本有差异,而在Ubuntu中可以通过切换cuda来满足需求,现记录如下。
1、按照 Ubuntu22.04与深度学习配置 中的cuda安装章节,将需要的cuda版本下载到本地并进行安装。
2、cuda安装完成后修改bashrc文件内容
sudo gedit ~/.bashrc
在最后添加以下命令
# cuda
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
保存并退出后,运行
source ~/.bashrc
3、进入/usr/local文件夹下,在控制终端下输入 ll
命令查看当前文件夹下的详细情况
可以看到我这里cuda文件夹是链接到了cuda-11.6文件夹。我们只要修改这个软链接就行,具体操作如下:
删除已经存在的软链接
sudo rm -rf cuda
建立新的软链接
sudo ln -s /usr/local/cuda-11.7 /usr/local/cuda
最终通过nvcc -V
命令查看当前系统的cuda版本