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
使其生效
完成!!