Windows下安裝tensorflow步驟
阿新 • • 發佈:2018-11-27
官方安裝連結
tensorflow 0.12.0開始支援Windows下安裝了
環境要求:
- Windows 64位
- python 3.5
- pip 9.0.1
- tensorflow 0.12.0
- cuda8.0
- cudnn5.1
- visual c++ 2015 執行時庫
檢視gpu是否支援cuda
安裝過程:
安裝python3.5
- 下載
- 安裝
- 升級pip
python.exe -m pip install --upgrade pip
安裝tensorflow
- gpu版本
pip install --upgrade tensorflow-gpu
- cpu版本
pip install --upgrade tensorflow
gpu版本tensorflow安裝gpu支援包
安裝cuda8.0
- 下載
- 安裝
安裝cudnn5.1
- 下載
- 安裝
新增解壓後的bin目錄到PATH環境變數
測試tensorflow
$ python
...
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!' )
>>> sess = tf.Session()
>>> print(sess.run(hello))
b'Hello, TensorFlow!'
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print(sess.run(a + b))
42
>>>
如果import報錯,需要下載安裝Microsoft Visual C++ 2015 Redistributable
另外需確認CUDA CUDNN與TensorFlow版本是否匹配
- gpu使用情況檢視
nvidia-smi -l # 實時返回GPU使用情況