CentOS 7 安裝 tensorflow

论坛 期权论坛 编程之家     
选择匿名的用户   2021-6-2 20:54   2241   0

官網上提供的懶人包似乎不適合CentOS系統 ,所以要install from source

1. 高版本gcc, python, java:已具備

$ java -version


2. 安裝bazel

問題出現在這裡。

如果直接下載bazel安裝包,仍舊會編譯失敗,與protoc之類(看不懂)有關,所以仍舊要install from source。

去網頁release page下載bazel-0.4.3-dist.zip,解壓後,檢查一下裡面所有文檔的權限(如果權限不可編譯,後續還會出現問題),只要確保權限可編譯,執行./compile.sh即可。

編譯的結果為output/bazel,執行

$cp output/bazel /usr/local/bin

完成。


3. Configuration

在tensorflow目錄下,執行./configure

除了Google Cloud Platform support 和OpenGL之外,選擇Yes或默認。

...

Configuration finished


4. Create the pip package and install

$ bazel build -c opt //tensorflow/tools/pip_package:build_pip_package

# To build with GPU support:
$ bazel build
-c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

$ bazel
-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

# The name of the .whl file will depend on your platform.
$ sudo pip install
/tmp/tensorflow_pkg/tensorflow-0.12.1-py2-none-any.whl

5. 设置环境参数

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"
export CUDA_HOME=/usr/local/cuda

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP