欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Ubuntu18.04安装cuda(10.1及10.0)和cudnn

程序员文章站 2022-07-05 18:17:12
...

Ubuntu18.04安装cuda(10.1及10.0)和cudnn

安装cuda10.1及其对应cudnn

先查询自己电脑的显卡驱动版本,命令如下:

nvidia-smi

Ubuntu18.04安装cuda(10.1及10.0)和cudnn上面的图是为安装完成之后的截图,不过不影响,只需要看一下自己的驱动版本找到适配版本即可。为安装的是cuda10.1,进入官方网站,选择合适的版本下载
cuda各版本网址链接:https://developer.nvidia.com/cuda-toolkit-archive
cudnn各版本网址链接:https://developer.nvidia.com/rdp/cudnn-archive
Ubuntu18.04安装cuda(10.1及10.0)和cudnn选择其中对应的版本,然后下载就行
下载之后,放在主目录下,打开终端,执行命令安装(文件名更换为所下载的):

sudo sh cuda_10.1.168_418.67_linux.run

安装选项,因为已经提前安装了显卡驱动,所以把显卡驱动的安装选项取消掉:
Ubuntu18.04安装cuda(10.1及10.0)和cudnn这里输入accept,回车
Ubuntu18.04安装cuda(10.1及10.0)和cudnn按下箭头,选择install,回车,出现如下界面,表示安装完成(不必担心警告,那是因为没有选择安装显卡驱动而出现的,忽略即可):
Ubuntu18.04安装cuda(10.1及10.0)和cudnn测试一下,输入命令:

cat /usr/local/cuda-10.1/version.txt

出现下图版本信息,表示成功

Ubuntu18.04安装cuda(10.1及10.0)和cudnn

配置cuda环境
打开用户配置文件:

sudo vim ~/.bashrc

可能是电脑版本不一样,所以命令也会有出入,我用上面的命令显示“sudo: vim: 找不到命令”,然后输入下面的命令:

sudo gedit ~/.bashrc

成功打开配置文件,在文件最后面加入如下语句:

export PATH="/usr/local/cuda-10.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/lcoal/cuda-10.1/lib64:$LD_LIBRARY_PATH"

保存,关闭,然后输入下面的命令使配置生效:

source ~/.bashrc

然后输入下面命令:

nvcc --V

查看安装版本,确认是否安装上:
Ubuntu18.04安装cuda(10.1及10.0)和cudnn下面是安装cudnn
cudnn是GPU加速计算深层神经网络的库。先去官网下载cudnn,不过需要先注册,然后下载对应版本的就行了。下载如下:
Ubuntu18.04安装cuda(10.1及10.0)和cudnn这里我选择的是cudnn7.6.5,上图上面那部分的最后一个,然后选择里面的for Linux版本,也就是上图下面部分的第三个,然后下载,我下载完成后将文件放在了主目录下。
然后输入下面的命令对下载后的文件进行解压:

tar -xvf cudnn-10.1-linux-x64-v7.6.5.32.tgz

复制cudnn内容到cuda相关文件夹内:

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h /usr/local/cuda-10.1/lib64/libcudnn*

cudnn版本查询命令:

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

出现如下图:
Ubuntu18.04安装cuda(10.1及10.0)和cudnn然后执行下面的命令来查看所安装的cuda和cudnn:

nvcc -V

出现如下图:
Ubuntu18.04安装cuda(10.1及10.0)和cudnn这里就表明安装成功了!

安装cuda10.0及其对应cudnn

下面开始cuda10.0及其对应的cudnn的安装
下载过程与前面过程类似,去NVIDIA官网进行下载(前面附有各个版本的cuda和cudnn)

cuda选择下图中的各个选项,然后下载。这里一共有两个包,一般只下载第一个包,第二个是patch,也就是补丁,可有可无。下载完成后放在主目录中。

Ubuntu18.04安装cuda(10.1及10.0)和cudnn
先执行下面的命令安装相关依赖环境:

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

下面执行命令进行安装:

sudo sh cuda_10.0.130_410.48_linux.run

然后下面是一堆协议说明,直接按q推出协议说明

Do you accept the previously read EULA?
accept/decline/quit: accept #接受协议

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: n #是否已有显卡驱动包,已有则选择n

Install the CUDA 10.0 Toolkit?
(y)es/(n)o/(q)uit: y #是否安装工具包,选择y

Enter Toolkit Location
 [ default is /usr/local/cuda-10.0 ]: #工具包安装地址,默认回车即可

Do you want to install a symbolic link at /usr/local/cuda?这里是建立软连接,若第一次安装cuda,就输入y;若之前安装过其他版本的cuda,则有两种情况,如果想启用当前安装的cuda版本就选y,如果只是安装这个版本就选n。

Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: n 
#**这里是建立软连接,若第一次安装cuda,就输入y;若之前安装过其他版本的cuda,则有两种情况,如果想启用当前安装的cuda版本就选y,如果只是安装这个版本就选n。**

Install the CUDA 10.0 Samples?
(y)es/(n)o/(q)uit: y

Enter CUDA Samples Location
 [ default is /home/zcc ]:  #样例安装地址默认即可
Logfile is /tmp/cuda_install_16509.log #**安装完成**

下面配置cuda10.0的环境变量,在终端输入下面的命令:

sudo gedit ~/.bashrc

然后会弹出一个界面,在界面的最后,另起一行,输入下面的命令:

export PATH=$PATH:/usr/local/cuda-10.0/bin  
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64  
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-10.0/lib64 

保存,退出,然后输入下面的命令使配置生效:

source ~/.bashrc

然后执行下面命令,查看安装版本,确认是否安装上:

nvcc --V

下面是安装cuda10.0对应的cudnn

下面是cudnn下载的链接选项,这里选择下面那部分的第三个即“cudnn library for Linux”,下载完成后放在主目录中。

Ubuntu18.04安装cuda(10.1及10.0)和cudnn
打开终端,先执行下面的命令进行解压:

tar -xvf cudnn-10.0-linux-x64-v7.6.2.24.tgz

然后执行:

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64
sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h /usr/local/cuda-10.0/lib64/libcudnn*

cudnn版本查询命令:

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

然后会出现:
Ubuntu18.04安装cuda(10.1及10.0)和cudnn到这里就安装成功了!