Ubuntu多版本CUDA切换--无需sudo权限

Ubuntu多版本CUDA切换--无需sudo权限


集群环境中,不同的用户往往需要配置不同的环境。CUDA作为当前深度学习必不可少的一个工具,其配置也是相当重要的。
假设我们安装了多个CUDA包,一般CUDA会安装在/usr/local里(当然也可以安装在任何地方,视自己的安装情况而定)

/usr/local/cuda-8.0
/usr/local/cuda-9.0
/usr/local/cuda-10.0

因为系统会默认使用下面路径的CUDA

/usr/local/cuda

这个CUDA其实只是上述CUDA的一个软链接(可以理解为windows里面的快捷方式),很多方法要求把这个路径给删除掉,然后创建一个所需CUDA版本的软链接。操作如下:

sudo rm -r /usr/local/cuda
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda

但是这样需要sudo权限,而集群环境中往往是没有sudo权限的,因此,我们可以通过修改环境变量的形式来完成,我们需要修改自己路径下的.bashrc文件(路径一般在~/.bashrc),打开.bashrc(用vim打开),增加下列内容:

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH 
export PATH=/usr/local/cuda-9.0/bin:$PATH  
export CUDA_HOME=/usr/local/cuda-9.0:$CUDA_HOME 

通过命令source ~/.bashrc使其生效

完成!!

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值