[Python知识库]Ubuntu20.04安装CUDA及cudnn

CUDA安装

1. 安装对应cuda(cuda-11.0)

   下载地址:https://developer.nvidia.com/cuda-11.0-download-archive
   

   # 下载链接以及安装命令
   wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
   sudo sh cuda_11.0.2_450.51.05_linux.run
   

   中间有部分选择:1.accept

   ? 2.cuda工具以及驱动下载;

   根据自己的情况下载

   成功安装完成之后:
   

   编辑环境配置:

   $ sudo gedit ~/.bashrc
   输入:
   # CUDA Soft Link
   export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
   export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
   

   查看是否成功:

   $ nvcc -V
   

2. 安装cudnn(cudnn v8.0.5)

   下载: https://developer.nvidia.com/rdp/cudnn-archive
   

   将下载的 cuDNN Library for Linux (x86_64) 解压，复制解压出来的文件到安装好的CUDA环境中

     sudo cp cuda/include/cudnn.h /usr/local/cuda/include
   sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
   sudo cp cuda/include/cudnn_version.h /usr/local/cuda/include
   # 更改权限输入： 
   sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* 
   

   # 安装结束后重启，测试是否安装成功
   # 方法1
   cp -r /usr/src/cudnn_samples_v8/ ~ 
   cd ~/cudnn_samples_v8/mnistCUDNN/
   make clean && make
   ./mnistCUDNN 
   

出现Test passed！没有报错即安装成功

      # 测试方法2
      cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
      sudo make -j8
      ./deviceQuery

出现 Result = PASS，说明安装成功

cuda和cudnn环境配置

# vi /etc/profile
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cudnn/lib64:$LD_LIBRARY_PATH
# 激活环境
source /etc/profile
# 验证环境是否正确
nvcc -V