ubuntu16.04 安装tensorflow1.4

来源:互联网 发布:变频器仿真软件 编辑:程序博客网 时间:2024/05/18 16:54

安装好基本环境后,cuda cudnn等

上这个网址 https://pypi.python.org/pypi/tensorflow-gpu 下载相应的whl文件

然后直接 

  1. sudo pip install tensorflow-1.4.0-cp27-cp27mu-manylinux1_x86_64.whl
便可安装成功
    

测试tensorflow

$ python

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> sess.run(hello)
'Hello, TensorFlow!'
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> sess.run(a + b)
42
>>>

使用该方法安装,需要定制的cudnn版本,可能会出现不兼容的情况

PIP 安装的优缺点

优点:应该是几种安装方法中最快的一种 
缺点:不能灵活定制,操作系统、GPU 硬件、CUDA 版本、cuDNN 版本必须与官方标称一致

如果本身带有的cudnn不匹配,如安装tensorflow_gpu-1.4.1-cp27-cp27mu-manylinux1_x86_64.whl,该版本要求的是libcudnn6.0,而我的电脑里面装有的是libcudnn7.0,如果运行import tensorflow,则会出现下面的错误:

  

因此,可以下载cudnn-8.0-linux-x64-v6.0.tgz,解压之后,将其中的libcudnn.so.6.0.20拷贝至/usr/local/cuda/lib64中,并使用sudo ln -s 添加软连接至libcudnn.so.6,便可以使用tensorflow了