Ubuntu16.04安装cuda9.1+cuDNN7.1简明教程

本文参考文章:
1)http://blog.csdn.net/jonms/article/details/79318566

最近因学习需要安装cuda,索性就装了最新版的cuda9.1,但安装过程中走了不少弯路(10系列显卡是真的有毒,被针对了两天)。今天终于装好了,写下来分享给大家,希望可以给大家提供一些参考~

我的环境:Win10+Ubuntu16.04双系统

首先推荐大家都去看下一NVIDIA官方给出的安装教程,可以避免很多不必要的问题(比如重装系统= =)

1.安装前准备

  • 验证系统是否安装了gcc。

gcc --version

  • 验证系统是否安装了正确的内核头文件和开发包。

sudo apt-get install linux-headers-$(uname -r)

2.安装NVIDIA驱动
这里我安装的驱动版本是390.25。我使用添加PPA的方式安装驱动,比较方便
PS:大家在安装驱动时最好先进入bios将系统的安全模式(security boot)关闭,避免出现显卡驱动装不上的情况。
(1)添加PPA

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

(2)搜索推荐的驱动

ubuntu-drivers devices

这里写图片描述
(3)选择recommened的驱动程序

sudo apt-get install nvidia-390

(5)重启系统

sudo reboot

(6)确认NVIDIA驱动是否装好

nvidia-smi

若驱动安装成功,应该可以看到如下输出:
这里写图片描述

3.禁用nouveau驱动
首先,使用如下命令在/etc/modprobe.d/blacklist-nouveau.conf中创建一个文件:

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

然后键入以下内容:

blacklist nouveau 
options nouveau modeset=0

接着重新生成内核:

sudo update-initramfs -u

重启计算机。然后在终端中输入lsmod | grep nouveau若没有输出,则表明禁用成功。

4.安装CUDA9.1
(1)找到CUDA9.1 toolkit所在位置,然后在终端中键入(我的是根目录):

sudo chmod +x cuda_9.1.85_387.26_linux.run 
sudo ./cuda_9.1.85_387.26_linux.run 

(2)安装过程中会有几个选项,如下:
注意:不要安装cuda自带的显卡驱动,其他选y,路径输入默认值就可以了。

accept/decline/quit: accept

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 387.26?
(y)es/(n)o/(q)uit: n

Install the CUDA 9.1 Toolkit?
(y)es/(n)o/(q)uit: y

Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y

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

(3)安装完成后
这里写图片描述
(4)添加路径
将路径添加在当前用户的profie里,操作如下:

sudo gedit /home/<用户名>/.profile 

<用户名>替换成自己的用户名即可
64位系统添加如下两行(32位系统将lib64改为lib即可):

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

之后(你没看错前面有个点)

source .profile

或者重启都可以使路径生效。
(5)验证CUDA是否安装成功
在终端中键入:

nvcc -V

可以看到cuda的版本信息
这里写图片描述

接着尝试运行一下CUDA中自带的例程,在/usr/local/cuda-9.1/samples/1_Utilities/deviceQuery中启动终端

cd /usr/local/cuda-9.1/samples/1_Utilities/deviceQuery
sudo make

然后

./deviceQuery

会有以下输出
(**注意:**出现的Result = Pass 则表明CUDA9.1已经完全安装成功~)
这里写图片描述

5.安装cuDNN
cuDNN链接:https://developer.nvidia.com/cudnn
需要先注册一个帐号,注册是完全免费的。

下载cuDNN v7.1.1 Library for Linux

然后将其解压复制到Cuda的目录中,并赋予其权限:

tar -xzvf cudnn-9.1-linux-x64-v7.tgz
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 /usr/local/cuda/lib64/libcudnn*

到这里cuDNN就已经安装成功了。

撒花花~~~奖励自己一顿大餐吧~

最后感谢一下参考文章的作者!!

  • 3
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值