Ubuntu 下不同版本 cuda 和 cudnn 切换

本文介绍了如何在Ubuntu系统中切换不同版本的CUDA和cuDNN。对于CUDA,重点在于确保环境变量指向软链接而非硬编码版本,通过修改软链接实现版本切换。而对于cuDNN,切换过程实际上是重新安装并创建软链接到对应的CUDA版本。详细步骤包括检查环境变量设置,创建和修改软链接等。
摘要由CSDN通过智能技术生成

Ubuntu 不同版本 cuda 和 cudnn 切换

可能不同的软件对与 cuda 和 cudnn 的版本有不同的要求,所以需要根据具体的要求来切换其版本。

ubuntu 是支持安装多个版本的 cuda 的,需要使用某个的版本时切换就可以了。对于 cudnn 则是直接重装,所谓的重装其实就是 copy 后创建个软链接就可以了。

安装 cuda 比较容易,直接运行下载的 sh 文件就可以了。

sudo sh cuda_9.0.176_384.81_linux.run

1. 切换 cuda

首先应该确保环境变量的设置不是 hard code,而是指向软链接,这样就可以直接通过修改软链接来修改 cuda 的版本。当然 hard code 的方式可以通过直接修改环境变量来修改 cuda 的版本,也是可以的。

通过如下命令进入环境变量文件

sudo gedit ~/.bashrc

检查关于 cuda 的环境变量设置,如下的设置是 hard code 的,直接指定了 cuda 的版本。

# for cuda9.0
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

而下面的设置是指向 /usr/local/cuda 的软链接。

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值