Windows10下安裝原生TensorFlow GPU版

  1. 下載 CUDA 8.01cuDNN v5.11python

  2. 安裝CUDAwindows

  3. 解壓cuDNN到一個你喜歡的位置,複製文件夾的絕對路徑並加到 PATH 環境變量去,而後把該文件夾下的bin文件夾的路徑也放到PATH裏去api

  4. 安裝Anaconda Python 3.5 version
    注意windows tf 僅支持 python 3.5google

  5. 安裝 tf
    爲了不以後使用 pip 安裝 tf 報錯,請先在cmd運行如下代碼code

pip install --ignore-installed --upgrade pip setuptools
以後安裝tf
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0-cp35-cp35m-win_amd64.whl
或者
pip install tensorflow-gpuip

接來下檢驗安裝是否成功get

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

原文做者: filwalinecmd

相關文章
相關標籤/搜索