Ubuntu 16.04 安装CUDA 9.0、cuDNN 7.05以及tensorflow-gpu教程

最新推荐文章于 2024-07-11 18:55:55 发布

逐梦南柯

最新推荐文章于 2024-07-11 18:55:55 发布

阅读量531

点赞数

本文链接：https://blog.csdn.net/weixin_43347844/article/details/86553461

版权

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))

没有报错说明安装成功