随笔-Ubuntu中多cuda版本环境配置
由于每个人做得任务不同，在实验室的服务器上，有多个用户和多个cuda版本，如何配置自己使用哪一个cuda呢？
看一下cuda的安装路径
- cuda一般安装在
/usr/local/目录下
配置要用的版本的CUDA的路径
- 配置自己使用的cuda版本，编辑
~/.bashrc
vi ~/.bashrc
- 插入以下配置。（想使用那个cuda版本就把配置中的cuda版本改成自己需要的那个）
# CUDA
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
- 激活更改
source ~/.bashrc
补充：
- 查看cuda的版本
# cat /usr/local/cuda/version.txt
cat /usr/local/cuda-10.1/version.txt
- 查看cuDNN的版本。（这里查出来的是7.6.2）
# cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda-10.1/include/cudnn.h | grep CUDNN_MAJOR -A 2
- 查看gcc的版本(5.4.0)
gcc --version
补充
- 有多个cuda版本，配置一个默认的cuda。创建一个软连接：
ln -s cuda-10.1 cuda
- 再看cuda版本：
stat cuda
创作不易，喜欢的话加个关注点个赞，❤谢谢谢谢❤
版权声明：本文为CSDN博主「快乐的小小程序猿」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/xu_benjamin/article/details/121479897
暂无评论