首页 > 技术文章 > Ubuntu18.04安装CUDA10、CUDNN

answerThe 2020-01-30 23:30 原文

我的系统是Ubuntu18.04、64位,选择CUDA10版本如下:

STEP1:CUDA官网选择适合自己系统的版本下载。

 

 STEP2:安装CUDA

在命令行中键入:

sudo sh cuda_10.2.89_440.33.01_linux.run

如果驱动是独立安装了,一定要选择不安装驱动!选择如下:

 

 安装完成后会提示如下:

这是因为我们在安装CUDA的时候没有选择安装驱动,提示需要安装驱动,忽略就行。

STEP3:添加环境变量:

export CUDA_HOME=/usr/local/cuda 
export PATH=$PATH:$CUDA_HOME/bin 
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存并退出

source ~/.bashrc

STEP4:测试是否安装成功

上面的选项选择安装了CUDA例子,运行其中一个来测试是否安装成功:

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

结果如下就表明成功啦

STEP5:安装CUDNN

官网下载安装包,需要注册登录才能下载。选择适合自己的版本 

 下载完成后解压并进入文件夹:

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*

在终端查看CUDNN版本:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

 查看cuda版本

cat /usr/local/cuda/version.txt

 

解决cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2无法查看cudnn版本的问题

关于cudnn版本的查看,大部分教程给的操作都是

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
但是新一点的cudnn都无法再通过这条指令查看版本号了

这是因为,新一些的cudnn版本信息都写在在cudnn_version.h而不是cudnn.h

所以复制h文件的时候,需要在

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 cp cuda/include/cudnn_version.h /usr/local/cuda/include/
查看版本号的指令也改为

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

推荐阅读