1. 程式人生 > >Hello TensorFlow 二 (GPU)

Hello TensorFlow 二 (GPU)

() pytho www. alt tel gef tro .org win

官方說明:https://www.tensorflow.org/install/

環境:

操作系統 :Windows 10 家庭中文版

處理器 : Intel(R) Core(TM) i7-7700 CPU @3.6GHZ 3.60GHZ

內存 :16GB

顯卡:NVIDIA GeForce GTX 1060 6GB

Python:3.6.2

安裝GPU版

pip3 install --upgrade tensorflow-gpu

技術分享

安裝 NVIDIA CUDA? Toolkit 8.0.

下載地址

https://developer.nvidia.com/cuda-downloads

技術分享

安裝 cuDNN v6

下載地址:https://developer.nvidia.com/rdp/cudnn-download

需要註冊用戶

技術分享

將下載後的zip文件解壓到c盤 C:\cudnn-8.0-windows10-x64-v6.0

將 C:\cudnn-8.0-windows10-x64-v6.0\cuda\bin 加入到環境變量Path

驗證tensorflow-gpu安裝

技術分享

執行sess = tf.Session()

後,會顯示顯卡GPU信息。

Hello TensorFlow 二 (GPU)