ubuntu系统上安装多个版本cuda/cudnn

人生简洁之道

已于 2022-01-26 16:35:48 修改

阅读量2.8k

点赞数 5

分类专栏： Mess 文章标签： ubuntu linux 服务器

于 2022-01-26 13:59:11 首次发布

Adress：CSDN - Life Recoder. PS: 爬虫、网站、公众号等侵权的当个人吧！乱写的笔记 1秒钟成了你的原创？

本文链接：https://blog.csdn.net/BeBuBu/article/details/122699243

版权

Mess 专栏收录该内容

42 篇文章 0 订阅

订阅专栏

nvidia cuda cudnn 的一些基本常识
- nvidia在ubuntu上面的驱动版本，没有对cuda有强约束，理论上安装一个较为新的版本即可，就可以识别显卡了。nvidia-smi
  - 查看显卡设备和驱动版本：ubuntu-drivers devices
  - 安装，挑选一个版本：sudo apt install nvidia-driver-440
  - 重启：reboot
- cuda和cudnn版本之间有对应关系：版本
  - 可以安装多版本cuda，并在各自的路径下加入对应版本的cudnn
  - 在～/.bashrc 中加入 cuda 的软连接路径，如下
  - 通过替换软连接指向来切换cuda版本
```
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin 
export LD_LIBRARY_PATH=/usr/local/cuda10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

# sudo rm -rf cuda
sudo rm -rf /usr/local/cuda
sudo ln -sf /usr/local/cuda-9.2 /usr/local/cuda

cd /usr/local/
stat cuda
```
cuda 10.0 安装流程

sudo chmod +x cuda_9.0.176_384.81_linux.run # 为 cuda_9.0.176_384.81_linux.run 添加可执行权限
sudo ./cuda_9.0.176_384.81_linux.run # 安装 cuda_9.0.176_384.81_linux.run

Do you accept the previously read EULA?
accept/decline/quit: accept

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?
(y)es/(n)o/(q)uit: n # 如果在这之前已经安装好更高版本的显卡驱动就不需要再重复安装，如果需要重复安装就选择 yes,此外还需要关闭图形界面。

Install the CUDA 9.0 Toolkit?
(y)es/(n)o/(q)uit: y

Enter Toolkit Location
 [ default is /usr/local/cuda-9.0 ]: # 一般选择默认即可，也可以选择安装在其他目录，在需要用的时候指向该目录或者使用软连接 link 到 /usr/local/cuda。

/usr/local/cuda-9.0 is not writable.
Do you wish to run the installation with 'sudo'?
(y)es/(n)o: y

Please enter your password: 
Do you want to install a symbolic link at /usr/local/cuda? # 是否将安装目录通过软连接的方式 link 到 /usr/local/cuda，yes or no 都可以，取决于你是否使用 /usr/local/cuda 为默认的 cuda 目录。
(y)es/(n)o/(q)uit: n

Install the CUDA 9.0 Samples?
(y)es/(n)o/(q)uit: n

cuda 10.0 -> cudnn 7.6.5 -> conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch CSDN
cuda 11.3 -> cudnn 8.2.1 -> conda install pytorch==1.10.1 torchvision torchaudio cudatoolkit=11.3 -c pytorch

sudo cp cudnn/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

查看cuda版本：nvcc -V
查看cudnn版本：
- 版本7：cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
- 版本8：cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2