Hello TensorFlow 二 (GPU)
阿新 • • 發佈:2017-09-24
() 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)