前置条件:已经安装好显卡驱动(2080TI),具体参考另外一篇博客。
安装环境:Ubuntu18.04 + CUDA10.1 + CUDNN7.6
一.安装CUDA10.1
1.版本选择
查看驱动满足的CUDA版本https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/
这里我们选择安装CUDA10.1
2.下载CUDA10.1
wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
3.安装CUDA
sudo sh cuda_10.1.243_418.87.00_linux.run
1)选择continue
2)输入accept
3)已经安装过显卡驱动,不需要再次安装,使用空格键取消勾选,然后选择install。
4)出现以下结果,则说明安装成功。
4.添加环境变量
1)配置CUDA的环境变量
vim ~/.bashrc ,在末尾添加如下内容
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1
2)使配置文件生效
source ~/.bashrc
5. 测试CUDA安装成功的Samples例子
依次执行如下命令
cd /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
Result = PASS 则说明安装成功
6. 查看CUDA版本
nvcc -V
二.安装CUDNN7.6
1.版本选择
选择与CUDA对应的CUDNN版本,官方下载地址:https://developer.nvidia.com/rdp/cudnn-archive ,需要注册登录。如下图所示:
2. 解压
tar –zxvf cudnn-10.1-linux-x64-v7.6.5.32.tgz
3. 执行如下命令
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
#重启系统
sudo reboot
4. 验证是否安装成功
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
出现CUDNN的版本信息,则说明安装成功。