在Ubuntu服务器搭建CUDA环境:ubuntu18.04.4 + CUDA10.2 + Cudnn8.0.0

初始环境:ubuntu18.04.4 LTS,已经安装对应的显卡驱动,安装过程中不建议更换apt-get国内镜像源

一、安装CUDA10.2

1、在nvidia 官网上下载对应的CUDA 版本,我们这里选择CUDA10.2 的版本,链接https://developer.nvidia.com/cuda-toolkit-archive。如图1.1 所示选择对应的系统版本和CUDA 版本下载至本地目录中,以下载到/home/einai/Downloads 为例。
注:可先下载到本地,然后在同一局域网中通过scp –r 指令传输到服务器的指定文件夹。

图1.1
2、下载完成后,在服务器的命令窗口中依次输入cdcd Downloadsls,查看是否存在cuda_10.2.89_440.33.01_linux.run 的文件。在该文件的目录下输入指令:

sudo sh cuda_10.2.89_440.33.01_linux.run

进行安装,输入本服务器的密码后会出现如图1.2 的画面。在最后一行输入accept 按下enter键,出现如图1.3 的画面。
在这里插入图片描述

图1.2

在这里插入图片描述

图1.3

如图1.3 所示,此时已经进入的CUDA 的安装界面,其中第一个选项Diver 是显卡驱动,因为本服务器已经安装显卡驱动,所以不再勾选,如果勾选安装会出现bug,慎重!其余4项默认安装,选择好后将光标移至’Install’处,按下enter 键,此时已经开始安装,切勿在命令窗口按下其它按键,等待即可。

3、安装完毕后,输入nvcc –V,如果出现下图1.4 所示,表明此时可能没有添加CUDA 的环境变量。在命令窗口依次输入:

cd gedit ~/.bashrc,

查看文件中是否有以下语句:

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

如果没有,添加进去,添加完毕后保存如图1.5 所示。此时依次输入指令:

source ~/.bashrc
nvcc –V

如果出现图1.6 所示画面,则表明CUDA 安装成功。
在这里插入图片描述

图1.4

在这里插入图片描述

图1.5

二、安装Cudnn

1、下载CUDA10.2 对应的Cudnn 版本,链接:https://developer.nvidia.com/rdp/cudnn-download。选择Cudnn8.0.0 版本进行下载,如图2.1 所示,选择圈出的三个deb 文件,高版本Cudnn 不建议使用tar 文件安装。这里依旧下载到/home/einai/Downloads 目录下。
在这里插入图片描述

图2.1

2、进入到deb 文件的目录,在此路径下依次输入以下指令,顺序不能颠倒

sudo dpkg -i libcudnn8_8.0.0.180-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.0.180-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn8-doc_8.0.0.180-1+cuda10.2_amd64.deb

3、(1)上述步骤已经完成Cudnn 的安装,现在需要验证Cudnn 是否安装成功。在窗口输入以下指令

cd
cd /usr/src

进入到src 文件夹下,使用ls 指令查看是否有‘cudnn_samples_v8’
文件夹。继续在src 目录下输入指令:

cp -r cudnn_samples_v8 /home/einai/Downloads/

将‘cudnn_samples_v8’文件夹复制到/home/einai/Downloads/下。

(2)在窗口依次输入以下指令:

cd
cd Downloads/cudnn_samples_v8/mnistCUDNN

进入mnistCUDNN 文件夹下,在此路径下继续输入指令make clean && make,执行结果如下图2.2 所示。
在这里插入图片描述

图2.2

(3)继续在mnistCUDNN 路径下输入指令:

./mnistCUDNN

执行结果如图2.3 所示,出现‘’test passed‘’表示Cudnn 安装成功。
在这里插入图片描述

图2.3

至此,基于ubuntu18的cuda环境已经配置完成!

  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值