切换cuda版本的两种方式

看正文之前,最好先搞懂什么是环境变量,PATH环境变量的作用,如何定义或修改环境变量,软链接的概念,什么是.bashrc文件,如何创建软链接,如何删除软链接。
参考:B站阿发你好的Ubuntu教程

名词解释:Linux系统中的软链接等价于Windows系统中的快捷方式


有两种切换cuda版本的做法,但两种方式是的本质是相同的,都是将cuda库的安装目录添加到PATH环境变量和LD_LIBRARY_PATH环境变量中。

方式1

不创建软链接,而是修改.bashrc中的环境变量指向;(如视频中讲的那样)

如图主目录安装好了5个版本的cuda,.bashrc文件中对应有上图5个版本的环境变量的定义,当前指向的是cuda11.0,切换cuda时只需更改注释即可。

方式2

不修改.bashrc文件,而是删除原有的cuda软链接,给想要的cuda库创建与被删除软链接的同名软链接;(前提是.bashrc文件中已经定义了环境变量的指向)

如上图,当前两个环境变量指向名为cuda的软链接,而这个软链接就是下边cuda10.1的快捷方式;这种情况下,可以不修改.bashrc文件,而是将名为cuda软链接删除,为你要使用的版本的cuda创建一个也名为cuda的软链接,这时候就指向新的版本的cuda了。

演示:

删除原来软链接

rm -rf 软链接名


给我想用的cuda(比如11.0)创建与被删除软链接同名的软链接

ln -s 源文件夹名 软链接名


查看下当前链接的cuda版本:

nvcc -V

总结

实战步骤

先打开.bashrc文件,查看环境变量的指向(图一还是图二类型),再做决定用第一种方便还是第二种方便。

再次强调

方式1是让环境变量直接指向cuda源文件,方式2是指向源文件创建的软链接(即快捷方式而已),两种方式本质都是指向原文件

你可能感兴趣的:(ubuntu,linux,python)