tensorflow-offline 离线安装

来源:互联网 发布:李喆工笔画网络班 编辑:程序博客网 时间:2024/06/01 07:29

tips:目前使用的是tensorflow1.4 版本,bazel 0.8.0版本,使用源码安装gpu版本

 

 

1.   第一步,下载相应的tensorflow的源代码

git clone https://github.com/tensorflow/tensorflow

 

2.安装bazel  

2.1 安装相关的依赖

sudo apt-get install build-essential openjdk-8-jdk python zip

 

sudo apt-get install pkg-config zip g++ zlib1g-dev unzip python 

 

2.2 下载bazel相关版本(bazel-0.8.0-installer-linux-x86_64.sh)

 

  2.3 安装bazel

chmod +x  bazel-0.8.0-installer-linux-x86_64.sh 

 

./bazel-0.8.0-installer-linux-x86_64.sh –user

 

vi ~/.bashrc

 

export PATH=”$PATH:$HOME/bin”

 

. ~/.bashrc

 

2.4 安装 bazel的相关包

sudo apt-get install python-numpy python-dev python-pip python-wheel

 

3.安装GPU的依赖

sudo apt-get install libcupti-dev

 

4.安装tensorflow

4.1 配置

cd tensorflow

./configure (tips:查看configure文件是否有x权限)


4.2 bazel 构建 pip 

注:由于是离线安装,tensorflow 中的WORKSPACE和 workspace.bzl两个文件中的包都要手动下下来,然后放在了本地的服务器上  eg:10.168.34.5

  

e.g:

  native.new_http_archive(

       name = "pcre",

       sha256 = "ccdf7e788769838f8285b3ee672ed573358202305ee361cfec7a4a4fb005bbc7",

       urls = [

           "https://mirror.bazel.build/ftp.exim.org/pub/pcre/pcre-8.39.tar.gz",

           "http://ftp.exim.org/pub/pcre/pcre-8.39.tar.gz",

       ],

       strip_prefix = "pcre-8.39",

       build_file = str(Label("//third_party:pcre.BUILD")),

   )

 

 

urls中的两个下载链接注释掉,然后换成本地服务器的下载链接。

 "http://10.168.34.5/workspace/pcre-8.39.tar.gz", 


把所有包都替换掉之后,执行以下命令就OK 啦。

bazel build –config=opt –config=cuda //tensorflow/tools/pip_package:build_pip_package



原创粉丝点击