一、安裝前的準備工作
在正式開始安裝TensorFlow之前,需要完成一些準備工作。
1、安裝Python環境,並且版本要在2.7.x或3.x之間,推薦使用Python3版本;
python --version
2、在Python環境中安裝pip包管理工具,可以使用系統包管理器或者官網提供的安裝腳本進行安裝;
sudo apt-get install python-pip
3、安裝OpenSSL開發包,這個是TLS/SSL的實現,默認情況下,TensorFlow會使用TLS/SSL進行數據傳輸,所以需要安裝OpenSSL來支持;
sudo apt-get install libssl-dev
二、使用pip安裝TensorFlow
安裝TensorFlow的最簡單方便的方式是使用pip命令安裝,該方法可用於Linux、Mac OS X和Windows等操作系統。
1、在終端或者命令行下輸入以下pip命令進行安裝:
pip install tensorflow
2、根據需要選擇不同版本的TensorFlow,例如,要安裝CPU版本的TensorFlow,可以使用以下命令:
pip install tensorflow-cpu
如果想要安裝GPU版本的TensorFlow,需要安裝相應的GPU版本的cuDNN和CUDA。
三、使用源代碼安裝TensorFlow
對於有一定編程基礎和一些運維經驗的用戶,可以選擇從源代碼編譯安裝TensorFlow。
1、首先,從GitHub上下載TensorFlow源代碼:
git clone https://github.com/tensorflow/tensorflow
2、安裝必要的依賴包,包括Python依賴、Bazel構建系統依賴、協議緩衝區編譯器等:
sudo apt-get install python-numpy python-dev python-pip python-wheel
sudo apt-get install bazel
sudo apt-get install protobuf-compiler
3、切換到tensorflow目錄下,使用bazel構建系統編譯TensorFlow:
cd tensorflow
./configure
bazel build --config=opt //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-version-tags.whl
四、安裝完畢後的測試
安裝完TensorFlow之後,可以使用Python的互動式界面或者編寫Python腳本進行測試。
1、在Python互動式界面中,輸入以下代碼,看是否可以正常導入tensorflow庫:
import tensorflow as tf
2、編寫一個簡單的腳本文件test.py,內容如下:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
在終端或者命令行下運行該腳本文件:
python test.py
如果輸出了「Hello, TensorFlow!」就代表安裝成功了。
原創文章,作者:BHZLA,如若轉載,請註明出處:https://www.506064.com/zh-tw/n/331347.html