Ubuntu16.04 pip下载安装tensorflow(GPU版)

来源:互联网 发布:期货实时行情数据接口 编辑:程序博客网 时间:2024/05/21 06:37

通过编译tensorflow编译源码再安装的方法(如本人博客Ubuntu16.04 源码安装tensorflow(GPU版) 中所讲),安装过的朋友就知道,这是个非常浪费时间的过程,最终编译生成一个whl文件,通过pip安装这个whl文件就行了。其实pip人家镜像源里面有一些常用版本的whl文件,不用咱们编译,直接下下来安装就可以了。
下面就介绍直接用pip下载安装tensorflow的方法。

cuda及cuDNN安装

这部分参见本人另外一篇博客Ubuntu16.04 Caffe安装笔记 中的一二三章节。
现在默认安装的tensorflow是1.4版本,要求安装cuda8.0和cudnn6

安装依赖

sudo apt-get install libcupti-devsudo apt-get install python-pip python-dev   # for Python 2.7sudo apt-get install python3-pip python3-dev # for Python 3pip install --upgrade pippip3 install --upgrade pip

安装tensorflow-gpu版本

根据自己用python2.7还是Python3来选择装相应的版本

pip install tensorflow-gpu  # Python 2.7pip3 install tensorflow-gpu # Python 3

测试

因为我安装的是python3版本

这里写图片描述


原创粉丝点击