1. 程式人生 > >TensorFlow初學者入門(一)——TensorFlow安裝測試

TensorFlow初學者入門(一)——TensorFlow安裝測試

本人學習TensorFlow中的一些學習筆記和感悟,僅供學習參考,有疑問的地方可以一起交流討論,持續更新中。

pip安裝TensorFlow

安裝 TensorFlow 過程中要涉及安裝或升級的包可以在官方的GitHub上獲得。

在Ubuntu或者Mac下安裝pip(python2)或者pip3(python3)的指令如下:

# Ubuntu/Linux 64-bit
$ sudo apt-get install python-pip python-dev

# Mac OS X
$ sudo easy_install pip

python2下安裝TensorFlow,注意區分系統以及有無GPU的要求:

# Ubuntu/Linux 64-bit, CPU only, Python 2.7:
$ sudo pip install --upgrade tensorflow
# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7. Requires CUDA toolkit 7.5 and CuDNN v4.
# For other versions, see "Install from sources" below.
$ sudo pip install --upgrade tensorflow

# Mac OS X, CPU only:
$ sudo easy_install --upgrade six
$ sudo pip install --upgrade tensorflow

python3下安裝TensorFlow,注意區分系統以及有無GPU的要求:

# Ubuntu/Linux 64-bit, CPU only, Python 3.4:
$ sudo pip3 install --upgrade tensorflow

# Ubuntu/Linux 64-bit, GPU enabled, Python 3.4. Requires CUDA toolkit 7.5 and CuDNN v4.
# For other versions, see "Install from sources" below.
$ sudo pip3 install --upgrade tensorflow
# Mac OS X, CPU only:
$ sudo easy_install --upgrade six
$ sudo pip3 install --upgrade tensorflow

測試是否安裝成功

進入python,執行

import tensorflow as tf

若無報錯,則安裝成功。