ubuntu16.04安装GPU版本的TensorFlow,以python2.7为例

1.安装pip

sudo apt-get install python-pip python-dev

2.由于我安装是cuda9.2和CuDNN7,所以选择从源码安装

如何安装cuda9.2,参考我的另外一篇文章

https://blog.csdn.net/williammms/article/details/80796798

安装CuDNN比cuda9.2简单很多,可以在网上找。

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*

3.安装了cuda9.2和CuDNN后,需要安装Bazel,在安装Bazel之前需要先安装JDK8。以下代码给出了安装JDK8的方法。

     1)sudo apt-get install software-properties-common

     2)sudo add-apt-repository ppa:webupd8team/java

     3)  sudo apt-get update 

     4)  sudo apt-get install oracle-java8-installer

4.下载Bazel,到  github.com/bazelbuild/bazel/releases  下载bazel-0.14.1-installer-linux-x86_64.sh,然后进行安装。

     1)chmod +x bazel-0.14.1-installer-linux-x86_64.sh

     2)  ./bazel-0.14.1-installer-linux-x86_64.sh   --user

     3)  export PATH="$PATH:$HOME/bin"

5.安装TensorFlow的依赖包

     sudo apt-get install python-numpy python-dev python-pip python-wheel

     pip install -U mock

      顺带如果pip版本低,可以升级一下pip。由于要使用GPU,要求cuda大于等于7.0,cuDNN大于等于v3,GPU计算能力大于3.0(到  developer.nvidia.com/cuda-gpus 对应查找)

6.配置TensorFlow

   到github.com/tensorflow/tensorflow 下载tensorflow或者

   git clone http://github.com/tensorflow/tensorflow

   解压后进入到tensorflow的文件夹

   ./configure

   第一个选择python的时候输入 python2.7

   第二个回车,第三个选择是否选用谷歌云,选择no,剩下参考如下:

         

7.配置完环境后,使用Bazel 来编译pip的安装包:

  1)bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_ pip_package

    2)bazel-bin/tensorflow/tools/pip_package/build_pip_package /tem/tensorflow_ pkg

    3)sudo pip2 install /tem/tensorflow_pkg/tensorflow-1.8.0-py2-none-any.whl

        在第一步耗时会很长,做好心理准备,其它两步,则没那么耗时。第三条指令的最后一个文件名不一定相同,需要自己用到那个文件夹去看,然后选择自己的whl文件来安装。

8.安装完毕

猜你喜欢

转载自blog.csdn.net/WILLIAMMMS/article/details/80811219