tensorflow 1.0 安装

来源:互联网 发布:淘宝虚拟店铺怎么入驻 编辑:程序博客网 时间:2024/06/08 06:28
教程在tensorflow 的官网上,我是用的源码编译
https://www.tensorflow.org/install/install_sources
首先安装 Bazel
也是源码编译
https://bazel.build/versions/master/docs/install-compile-source.html
在编译bazel之前把PATH加上JDK的二进制路径比如$HOME/java/jdk8xxxxx/bin就好,编译脚本会自动找到java然后编译,编译好之后直接把bazel二进制文件从 bazel-bin/src/bazel 拷贝到PATH里面就好,之后tensorflow只会用到这个二进制文件;
之后 下载 bazel-<VERSION>-dist.zip
unzip 之后, 执行 bash ./compile.sh 会自动编译,
编译结束后, 将生成的bazel-bin 的添加到你的PATH 环境变量里面去

此时,安装tesorflow 的依赖, numpy, dev, pip, wheel

然后进入到tensorflow 文件夹下, 执行./configure 然后按照提示, 一步一步选择, Y/N

之后, 需要安装tensorflow 自己的pip 包
还是在tensorflow的目录下, 执行
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
这个是支持gpu版的
大概需要十几分钟左右,装完之后, 执行下面命令
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
此时, 会在/tmp/tensorflow_pkg 这个目录下生成一个.whl 的文件
执行
pip install /tmp/tensorflow_pkg/xxx.whl
此时如果不报错, 就证明安装成功了.

进入到examples/mnist,
跑一下mnist_softmax.py 看看能否成功运行.


0 0
原创粉丝点击