Ubuntu下安装cuda和cudnn的方法

1.cuda的安装

(1)下载安装文件。首先去英伟达官网下载cuda安装包:https://developer.nvidia.com/cuda-toolkit-archive

按照系统配置选择安装包,据说用deb文件安装容易出错,所以直接下载runfile(后缀为.run)文件来安装。
Ubuntu下安装cuda和cudnn的方法_第1张图片
(2)安装cuda。cuda8.0和cuda9.0都集成了nvidia显卡驱动,按照安装cuda的同时是否安装显卡驱动可以分为不带驱动的安装方法和带驱动的安装方法。

1.1 最简单的方法——分开安装驱动和cuda

(1)更新系统的驱动。

系统设置-》软件和更新-》附加驱动-》更新nvidia和intel

假如更新驱动出现了循环登录系统的问题,请先卸载驱动,然后使用带驱动的安装方法来解决问题。

此时驱动的卸载方法:由于登陆进入不到图形用户界面(GUI),但我们可以进入到文本用户界面(TUI)

在登陆界面状态下,按Ctrl + Alt + f1,进入TUI执行

sudo /usr/bin/nvidia-uninstall

然后重启

sudo reboot
(2)安装cuda。

执行如下命令:

sudo sh cuda_9.0.176_384.81_linux.run

接着按下面的步骤选择:

accept

n(不要安装driver)

y

y

y

安装完成后,设置环境变量。

(3)设置环境变量。

打开主目录下的 .bashrc文件添加如下路径,例如.bashrc文件在/home/XXX下,如果没有找到,则按Ctrl+H键显示隐藏文件。

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

终端运行:source ~/.bashrc

检查:nvcc --version,如果显示下面的文子就说明安装成功了。
在这里插入图片描述

2.cudnn的安装

cudnn的安装非常简单,

(1)下载安装文件

按需求下载cudnn的安装文件:https://developer.nvidia.com/rdp/cudnn-archive
Ubuntu下安装cuda和cudnn的方法_第2张图片
注意:下载cuDNN v6.0 Library for Linux

(2)安装cudnn

解压下载的文件,可以看到cuda文件夹,在当前目录打开终端,执行如下命令:

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*
(3)查看cudnn版本

在终端输入

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

如果出现下图所示版本信息,说明安装成功。
在这里插入图片描述

你可能感兴趣的:(Ubuntu下安装cuda和cudnn的方法)