1.查看自己系统了安装了哪些cuda,后面切换cuda版本时,只能切换已经安装过的cuda版本。

ls -l /usr/local | grep cuda

可以看到,我的系统中安装了cuda11.1、11.3、11.6、11.8等版本

2.通过环境变量直接切换cuda版本

2.1 终端输入以下指令查看自己的cuda版本

nvcc -V

可以看到,我的cuda版本目前是11.1。

2.2 修改bashrc 文件切换cuda版本

2.2.1打开bashrc文件

sudo vi ~/.bashrc

找到下面三行代码,将自己的cuda版本改成所需的版本(前提是系统已经安装所需的cuda),如找不到,则将下面三行代码添加到文件最后,然后在修改。例如我将cuda版本从11.1切换到11.8。

export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
export LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LIBRARY_PATH

修改前:

修改后:

注意:打开文件后,按 i 进入编辑模式,修改后,按下电脑ESC 按键退出编辑模式,输入“:wq”  后按Enter 保存退出。如不想保存文件,则输入”:q!“按Enter 强制不保存退出

2.3 更新配置文件

source ~/.bashrc

2.4 再次查看cuda 版本

nvcc -V

可以看到,cuda版本已经切换到11.8了

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐