ubuntu14.04 安装anaconda+tensorflow-gpu版本

方法一：针对Ubuntu自带的python进行的安装（需要在安装anaconda前安装）：
一.安装pip

bisheng@bisheng:~$ sudo apt-get install python-pip

二.安装Nvidia Driver
1.查看自己的显卡版本!
这里写图片描述
我的是384

bisheng@bisheng:~$ sudo add-apt-repository ppa:graphics-drivers/ppa
bisheng@bisheng:~$ sudo apt-get update && sudo apt-get install nvidia-384

(这里384是因为我的型号是384)
出现问题就试试 –fix-missing
安装后，可以在这里查看
这里写图片描述
查看这里，图片中左边一列也变得很详细

三.安装CUDA
1. 命令行输入conda search tensorflow-gpu查看cuda,cudnn,python,以及tensorflow-gpu版本的对应关系

2.https://developer.nvidia.com/cuda-80-ga2-download-archive
选择CUDA版本

下载以后：

bisheng@bisheng:~$ sudo dpkg -i cuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64.deb
bisheng@bisheng:~$ sudo apt-get update
bisheng@bisheng:~$ sudo apt-get install cuda
bisheng@bisheng:~$ export PATH=/usr/local/cuda-8.0/bin/{PATH:+:${PATH}}
bisheng@bisheng:~$ export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64/{LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

bisheng@bisheng:~$ sudo gedit ~/.bashrc

之后在bashrc里添加两行：

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

我下载的是CUDA8.0，下载完以后查看CUDA:
这里写图片描述
查看之前的conda search tensorflow-gpu得到的表
我选择tensorflow-gpu 1.2.1 py27cuda8.0 cudnn5.1_0 pkgs/free

三.安装CUDNN
https://developer.nvidia.com/rdp/cudnn-archive
1.下载以后解压，再运行以下命令：

bisheng@bisheng:~$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
bisheng@bisheng:~$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
bisheng@bisheng:~$ sudo chmod a+r /usr/local/cuda/include/cudnn.h
bisheng@bisheng:~$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

2.运行检查是否安装成功：

bisheng@bisheng:~$ nvcc -V

这里写图片描述
3.重启
4.执行例程，查看CUDA版本

bisheng@bisheng:~$ cd /usr/local/cuda/samples/1_Utilities/deviceQuery
bisheng@bisheng:~$ sudo make
bisheng@bisheng:~$ ./deviceQuery

最后一行显示PASS就说明CUDA安装成功

四.安装GPU版的tensorflow

bisheng@bisheng:~$ pip install tensorfolw-gpu==1.2.1

执行tensorflow

bisheng@bisheng:~$ python
bisheng@bisheng:~$ import tensorflow as tf

如果此处报错可能tensorflow 版本和python版本不匹配，安装python对应版本使之与tensorflow匹配或者直接重装tensorflow
这里写图片描述
能正常import tensorflow库,那么
安装成功！

方法二：在Anaconda中创建虚拟环境进行安装
(注意：卸载该虚拟环境或者anaconda以后tensorflow,cuda,cudnn等也会自动被一同卸载)
一.安装Anaconda
1.选择anaconda python2.7版本
2.安装完毕后，在命令行中输入：

bisheng@bisheng:~$ sudo gedit ~/.bashrc

在bash 文件中添加行：

export  PATH=/home/bisheng/anaconda2/bin:$PATH

(依据anaconda安装的路径进行更改）
二.安装tensorflow

bisheng@bisheng:~$ conda create -n py35 python=3.5 tensorflow-gpu=1.8
bisheng@bisheng:~$ source activate py35

这里写图片描述

ubuntu14.04 安装anaconda+tensorflow-gpu版本

猜你喜欢