Ubuntu16 安装gpu模式的tensorflow

1. 准备工作

确认自己的电脑上安装了nividia驱动,以及cuda,cudnn,具体安装方法在另一片文章中有介绍

2. pip安装tensorflow

ubuntu16自带的Python2.7 和Python3.4,最好使用python3,运行

pip3 install tensorflow-gpu

可能的坑

ImportError:libcudnn.so.6.0:cannot open shared object file:No such file or directory
打开/usr/local/cuda/lib64,发现现存的文件为libcudnn.so.5.0
tensorflow-gpu最新版本是1.3,需要cuda8以及cudnnv6的支持,而笔者的电脑上安装的为cuda8和cudnnv5,如果不想更新cudnn,就需要安装gensorflow-gpu1.2版本

pip3 uninstall tensorflow-gpu
pip3 install tensorflow-gpu==1.2

再次打开python3,输入import tensorflow,就不会报错了

3. 最后提醒

不要使用源码安装!不要使用源码安装!不要使用源码安装!
笔者使用源码安装了三遍,折腾了一整晚,中间各种坑,最后也没安装成功。最好使用pip安装,一条命令搞定。是有多跟自己过不去非得用源码安装。

4. 更新

首先,感谢天瑞老哥,用了整整一中午帮我解决问题。
这一次,尝试在anaconda_python2.7环境中配置tensorflow.

1)安装anaconda

在conda官网下载对应python2.7版本的安装包,或前往镜像网站下载对应版本
[]!https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
python2.7对应Anaconda2-4.3.1

bash 安装包路径

添加路径

echo ‘export PATH=”~/anaconda2/bin:$PATH”‘ >> ~/.bashrc
source ~/.bashrc

2)conda环境管理

创建python2.7的运行环境,这里默认安装python2.7,如安装python3.4则需加上python==3.4

conda create –name tensorflow_py27

安装成功后,激活conda环境

source activate tensorflow_py27

3)安装tensorflow

pip install tensorflow_gpu

安装完成进入python后,发现问题来了:

File “/home/*/.local/lib/python2.7/site-packages/tensorflow/python/client/client_lib.py”, line 35…
ImportError: libcudart.so.7.0: cannot open shared object file: No such file or directory

之后,无论笔者安装哪个tensorflow版本,都会报这个错误。
原因出在.local上,发现此时并没有在conda文件夹下运行,这说明之前在本地可能安装过另一个版本的tensorflow。在home目录下执行

pip uninstall tensorflow

原来n年前安装过tensorflow0.5。。。
好啦,之后再在conda下安装tensorflow就没问题了

    原文作者:nowherespyfly
    原文地址: https://www.jianshu.com/p/f9a029ec3b41
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞