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

  • 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

你可能感兴趣的:(Mess,ubuntu,linux,服务器)