Ubuntu22.04切换系统cuda版本

由于最近项目要求的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命令查看当前文件夹下的详细情况
Ubuntu22.04切换系统cuda版本_第1张图片
可以看到我这里cuda文件夹是链接到了cuda-11.6文件夹。我们只要修改这个软链接就行,具体操作如下:
删除已经存在的软链接

sudo rm -rf cuda

建立新的软链接

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

最终通过nvcc -V命令查看当前系统的cuda版本

Ubuntu22.04切换系统cuda版本_第2张图片

你可能感兴趣的:(Ubuntu,深度学习,python,ubuntu,深度学习)