比思論壇

標題: Ubuntu16.04安装tensorflow_gpu教程 [打印本頁]

作者: naturo    時間: 2018-9-4 21:47
標題: Ubuntu16.04安装tensorflow_gpu教程

一,安装pip

命令行输入:sudo apt-get install python-pip python-dev

可能会出现错误,如无法定位软件包 python-pip  软件包python-dev 没有可安装候选,这时候需要更新下软件包信息,

具体做法是在terminal终端下执行 sudo apt-get update

再执行上一句

二,安装nvidia driver

驱动版本推荐:ubuntu-drivers devices

我的输出:

vendor   : NVIDIA Corporation

model    : GM107M [GeForce GTX 960M]

modalias : pci:v000010DEd0000139Bsv000017AAsd00003802bc03sc02i00

driver   : xserver-xorg-video-nouveau - distro free builtin

driver   : nvidia-384 - distro non-free recommended

所以安装命令:sudo apt-get install nvidia-384

三,安装cuda 8.0

(1) 注意作者给的不是8.0版,下载8.0请移步点击打开链接, 或者百度云:点击打开链接

(2) 找到下载目录,在下载目录打开terminal,根据下载的文件名相应的输入安装命令,我的文件名是:8.0.61_375.26_linux.run

所以安装命令为:

sudo sh cuda_8.0.61_375.26_linux.run --override

(3) 配置cuda环境变量方式错误,按作者方式,将导致笔记本重启后,cuda安装失败。正确方式如下:

    1)打开bashrc文件:sudo gedit ~/.bashrc

    2) 在文件最后输入:

    3)其次在bashrc文件最后添加一行依赖库:


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


    4)重启电脑,再输入:nvidia-smi

四,安装cudnn 5.1

(1) 作者给的同样不是5.1版,下载5.1请移步点击打开链接,需要注册,或者百度云:点击打开链接

安装过程就是解压:

tar xvzf cudnn-8.0-linux-x64-v5.1.tgz

然后里面的文件拷贝到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 /usr/local/cuda/lib64/libcudnn*

五,安装tensorflow-gpu版

1. 可以直接安装,输入:sudo pip install tensorflow-gpu==1.0

2. 也可以按照作者的方式,不过他的不完整,且版本过低。

作者给的是不完整的命令,正确的命令是:(1)export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

(2)sudo pip install --upgrade $TF_BINARY_URL

可能也会出错,试试多输入几遍,我试了四遍成功了。

Successfully installed funcsigs-1.0.2 mock-2.0.0 numpy-1.14.1 pbr-3.1.1 protobuf-3.0.0 setuptools-38.5.2 six-1.11.0 tensorflow-0.11.0 wheel-0.30.0

安装成功








歡迎光臨 比思論壇 (http://108.170.10.234/) Powered by Discuz! X2.5