LINUX下安装tensorflow

2018-12-19

linux(centos)下进行tensorflow的CPU版本安装,python版本是3.6,使用的方法是pip

1.安装python

  到目前为止,TensorFlow暂时只支持python3.4,3.5和3.6的,还支持python2.x。官网见这里。

  如果安装了python3.7的话,可以直接安装python3.6,不需要对3.7做处理,安装方法见之前的博客

2.安装pip

  一般情况下,安装完Python3.6后,会成功安装上pip,可以通过下面的代码来检查是否安装成功

[root@localhost ~]# pip --version
pip 18.1 from /usr/local/lib/python3.6/site-packages/pip (python 3.6)

  可能pip不是最新版,此时需要更新一下,免得后面使用pip来安装TensorFlow时出错。但更新的时候会报错,说不支持SSL,所以这要如下操作下

[root@localhost Python-3.6.7]# yum install openssl-devel -y   
[root@localhost Python-3.6.7]# cd /server/tools/Python-3.6.7/ # 进入到python的安装目录处,每个人的安装目录都不一样
[root@localhost Python-3.6.7]# make clean
[root@localhost Python-3.6.7]# ./configure [root@localhost Python-3.6.7]# make && make install [root@localhost Python-3.6.7]# python -m pip install --upgrade pip

3.安装TensorFlow

无论是安装CPU版本还是GPU版本,如果遇到timeout问题,可以改变pip的timeout时间。

pip --default-timeout=1000 install -U pip
  • TensorFlow-CPU版本

  接下来使用pip命令来安装TensorFlow即可,此版本是CPU版本,下载时间会有点久。

pip3 install --upgrade tensorflow #CPU版本
pip3 install --upgrade tensorflow #GPU版本,选择一个即可

  安装完后,进入Python编程,检测是否成功

Python 3.6.7 (default, Dec 19 2018, 12:25:37) 
[GCC 4.8.5 20150623 (Red Hat 4.8.5-36)] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow
>>>

  若没有错误消息,则说明安装成功

  • TensorFlow-GPU版本

由于TensorFlow下载是从国外服务器下载的,就算修改了timeout时间,可能也会遇到下载慢的问题。

此时,可以将TensorFlow换成国内源。

pip install -i https://pypi.mirrors.ustc.edu.cn/simple/ tensorflow-gpu==1.5.0

 此外,国内源还有:

阿里云 http://mirrors.aliyun.com/pypi/simple/ 

中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/ 

豆瓣(douban) http://pypi.douban.com/simple/

清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/ 

中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/

其中,清华的源貌似停止服务了。

特别要说一下,在上面命令中“tensorflow-gpu==1.5.0”,1.5.0是TensorFlow-GPU的版本号,这个版本号需要根据自身的显卡相结合,且需要搭配合适的cudnn和cuda,下图是对应的信息。

具体安装cudnn和cuda的方法,可以参考这个网站

安装完cudnn和cuda后,进入python,并且“import tensorflow”无警告,即安装成功。

  • 如果在使用过程中,出现“ImportError: libcublas.so.10.0: cannot open shared object file: No such file”的错误信息,说明TensorFlow和cuda的版本不对应,需要重新安装。
  • 如果出现“could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR”的错误,则解决方法如这个网站所示。
原文地址:https://www.cnblogs.com/dengshunge/p/10143341.html