CUDA安装

安装对应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

在这里插入图片描述

安装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