1. 程式人生 > >TensorFlow學習筆記(三)—— linux安裝TensorFlow

TensorFlow學習筆記(三)—— linux安裝TensorFlow

參照官方文件https://www.tensorflow.org/get_started/os_setup.html
我們可以使用pip來安裝tensorflow
因為依賴python,且有python2與python3之分,所以先安裝virtualenv
可方便得在python2.7與python3之前動態切換
sudo pip install virtualenv


啟用環境
virtualenv --system-site-packages ~/tensor
cd ~/tensor/
bash的話
source bin/activate
如果是csh則source bin/activate.csh
注意,這一步每次要執行tensorflow時都要執行的


這裡pip與pip3有區別的,我用的python3的版本
注意,如果是gpu的還要安裝CUDA9.0工具包,與 CUDA 工具包 9.0 相關聯的 NVIDIA 驅動程式,cuDNN v7.0, libcupti-dev 庫
sudo pip3 install --upgrade tensorflow-gpu
如果不帶gpu則是
sudo pip3 install --upgrade tensorflow
如果是python2則把上面的pip3改成pip


退出環境
deactivate


要缷載掉tensorflow, 這後面的tensorflow-gpu要對應前面裝的,如果是gpu的後面帶-gpu
sudo pip3 uninstall tensorflow-gpu


原始碼編譯tensorflow
安裝java-1_8_0-openjdk
安裝bazel 這個可以從原始碼或者直接系統安裝
如果要編譯bazel則要安裝java 8 jdk,可以是openjdk-8 其他版本的不行
原始碼方法:

https://github.com/bazelbuild/bazel/releases
下載最新的版本,我下的是bazel-0.13.0-dist.zip
mkdir bazel
cd bazel
unzip -x bazel-0.13.0-dist.zip
bash ./compile.sh
可以參考官方的https://docs.bazel.build/versions/master/install.html


git clone https://github.com/tensorflow/tensorflow 
cd tensorflow
./configure
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
sudo pip install /tmp/tensorflow_pkg/tensorflow-YOUR-VERSION.whl
上面的--config=cuda 去掉就是編譯CPU版


檢驗是否安裝正常:
(tensor)

[email protected]:~/tensor> python3
Python 3.6.5 (default, Mar 31 2018, 19:45:04) [GCC] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> 


其他參考
https://blog.csdn.net/wenanshi/article/details/73484688
https://blog.csdn.net/u012436149/article/details/53312212
http://vividfree.github.io/%E5%B7%A5%E5%85%B7%E5%AE%89%E8%A3%85/2017/04/13/install-tensorflow-on-linux-from-sources
http://wiki.jikexueyuan.com/project/tensorflow-zh/get_started/os_setup.html