Ubuntu16.04下CUDA 9.0 + cuDNN v7.0 + tensorflow 1.6.0（GPU）环境搭建

由于自己攒了个主机，第一次安装GPU版本的tensorflow，mark一下。

说明一下，本篇上接《Ubuntu16.04LTS下搭建强化学习环境gym、tensorflow》这篇文章，只不过修改了第四步（安装tensorflow）。

说一下环境的版本：

系统：Ubuntu 16.04.3

显卡：gtx 1080（索泰至尊Plus OC）

CUDA 9.0

cuDNN v7.0.5 for CUDA 9.0

tensorflow 1.6.0 for python 3.5.4

步骤如下：

1.安装显卡驱动

sudo apt-get update
sudo apt-get upgrade
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-367
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev

重启一下，使驱动生效，并使用以下命令测试驱动是否安装成功。

nvidia-smi

2.安装CUDA

官网链接，在这里下载好CUDA后，使用以下命令安装。

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local-rc_9.0.103-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local-rc/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

修改环境变量并使其生效。

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

这时，显卡驱动可能会被覆盖，重启一下万无一失。

3.安装cuDNN

官网链接，在这里下载好cuDNN，使用以下命令安装。

tar -zxvf cudnn-9.0-linux-x64-v7.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

4.安装tensorflow

先激活gymlab

source activate gymlab

安装python3-pip和python3-dev

sudo apt-get install python3-pip python3-dev

由于我想安装最新版本的支持python3.5的tensorflow1.6.0，然后我尝试使用以下命令进行安装。

sudo pip3 install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-1.6.0-cp35-cp35m-linux_x86_64.whl

报错了，404 not found，说明没有这个源，所以我又使用了以下命令进行安装。

pip3 install tensorflow-gpu

这次没报错，而且下载的正是最新版的1.6.0，也正是支持python3.5的。难不成这条命令会根据本机的python版本来下载最新版本的tensorflow？这个问题待定。

Ubuntu16.04下CUDA 9.0 + cuDNN v7.0 + tensorflow 1.6.0（GPU）环境搭建相关推荐

GTX960M安装Anaconda+cuda9.0+cudnn v7.6.5+tensorflow-gpu1.8.0
目录 1 安装Anaconda 1.1下载Anaconda安装包 1.2安装 1.3 更改路径 1.4 修改默认浏览器 2 安装cuda9.0 2.1 cuda版本选择 3 安装cudnn v7.6. ...
ubuntu16.04（虚拟机）上MPD（Music Player Daemon）+MPC运行环境搭建
MPD播放器为C/S架构,MPD为Server,它的Client 有很多,MPC为其典型稳定的Client. 1.源码下载在github 上有其源码,下载地址:https://github.com/ ...
ubuntu16.04下 1080ti显卡驱动384.98+cuda9.0+cudnn+caffe 安装过程，本人新测，没毛病
ubuntu16.04下显卡驱动384.98+cuda9.0+cudnn9.0+caffe 安装我自己尝试的在16.04下安装如上所述,系统安装不多说,主要说显卡驱动往下的部分,和我自己遇到的问题 ...
Ubuntu16.04 安装 CUDA、CUDNN、OpenCV 并用 Anaconda 配置 Tensorflow 和 Caffe 详细过程（此种方案不好，好的方案是另一篇，基于pyhton2的）
Ubuntu16.04 安装 CUDA.CUDNN.OpenCV 和 Caffe 详细过程(基于Python2,没有anaconda2和3,因为基于anaconda2和3在安装caffe时会报错,很难 ...
Ubuntu16.04 安装 CUDA、CUDNN、OpenCV 并用 Anaconda 配置 Tensorflow 和 Caffe 详细过程
Ubuntu16.04 安装 CUDA.CUDNN.OpenCV 并用 Anaconda 配置 Tensorflow 和 Caffe 详细过程 1.配置环境 Ubuntu16.04 64-bit系统 ...
ubuntu16.04安装CUDA 8.0（很详尽，包括一些坑的解决方法）
ubuntu16.04安装CUDA 8.0(很详尽,包括一些坑的解决方法) 参考文章: (1)ubuntu16.04安装CUDA 8.0(很详尽,包括一些坑的解决方法) (2)https://www. ...
Ubuntu16.04下TensorFlow-GPU安装记录（GTX1060显卡）
Ubuntu16.04下TensorFlow-GPU安装记录 1 确定电脑型号(重要的是显卡型号) 2 确定CUDA版本.ubuntu版本 3 Win10系统下安装Ubuntu组成双系统(简述过程) ...
win10 + NVIDIA GeForce RTX 2080 Ti + CUDA10.0 + cuDNN v7.6.5
win10 + NVIDIA GeForce RTX 2080 Ti + CUDA10.0 + cuDNN v7.6.5 配置教程参考文献: 本文参考了一下文献: 显卡驱动版本号一定要与cuda版本 ...
Ubuntu16.04下caffe CPU版的详细安装步骤
一.caffe简介 Caffe,是一个兼具表达性.速度和思维模块化的深度学习框架. 由伯克利人工智能研究小组和伯克利视觉和学习中心开发. 虽然其内核是用C++编写的,但Caffe有Python和Mat ...

Ubuntu16.04下CUDA 9.0 + cuDNN v7.0 + tensorflow 1.6.0（GPU）环境搭建

Ubuntu16.04下CUDA 9.0 + cuDNN v7.0 + tensorflow 1.6.0（GPU）环境搭建相关推荐

最新文章

热门文章