1. 程式人生 > >Windows下安裝tensorflow步驟

Windows下安裝tensorflow步驟

官方安裝連結
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

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使用情況

原文連結