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)
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