Ubuntu 16.04 安装CUDA 9.0以及cuDNN 7.05教程 （免入坑）

第一步：CUDA安装

下载CUDA

在NVIDIA官网上下载cuda 9.0,点击

link

.或者在官网搜索中输入CUDA 9.0 ,在搜索项中选择CUDA Toolkit Archive,在页面选择

linix

→

x86_64

→

ubuntu-deb


选择CUDA9.0

安装CUDA

在下载好的

deb文件目录

下打开终端，在命令行中输入(注意：原网站中第二命令的改成 9-0-local)：

//CUDA安装命令
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

设置环境变量以及路径

在任意终端命令行中输入：

// 设置环境变量以及路径命令
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} 

第二步：安装cuDNN

下载cuDNN

点击链接

link

, 下载对应版本的cudnn, 注意cudnn要和cuda 版本对应上


注意

：下cuDNN时需要注册英伟达的账号才能下载

安装cuDNN

选择其中的三个deb文件下载，将三个文件放在任意文件夹下，在

该文件夹

中打开终端并依次输入以下指令：

// 安装cuDNN命令
sudo dpkg -i libcudnn7_7.0.5.15-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.0.5.15-1+cuda9.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.0.5.15-1+cuda9.0_amd64.deb

第三步：安装tensorflow1.9.0

在终端中输入指令:

// tensorflow-gpu 1.9.0安装命令
sudo pip install tensorflow-gpu==1.9.0

然后重启电脑（有时候能解决很多问题）

第三步：验证安装

打开python输入以下代码：

import tensorflow as tf
a = tf.constant("hello world")
sess = tf.Session()
print(sess.run(a))

没有报错说明安装成功