Windows10下安裝原生TensorFlow GPU版

  1. 下載 CUDA 8.0cuDNN v6 for CUDA 8.0

(下載cuDNN須要先註冊NVIDIA開發帳戶並登陸才能看到下載界面)python

CUDA 9 要TF 1.5版本才支持windows

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

注意windows tf 僅支持 python 3.5code

  1. 安裝 tf

爲了不以後使用 pip 安裝 tf 報錯,請先在cmd運行如下代碼
pip install --ignore-installed --upgrade pip setuptools
以後安裝tf
pip install tensorflow-gpu==1.1.0ip

接來下檢驗安裝是否成功開發

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

另外,若是用conda的話必定要注意python版本,目前最高支持3.5get

conda create --name tensorflow python=3.5
activate tensorflow
pip install tensorflow-gpu

原文做者: filwalinecmd

相關文章
相關標籤/搜索