Windows10上安裝 TensorFlow1.4 + CUDA8.0 + cudnn6.0

安裝TensorFlow 1.4版本 的環境要求 如圖所示:

這裏寫圖片描述

安裝3.5以上版本的python,推薦直接安裝Anaconda:https://www.continuum.io/downloads

CPU版本:

  • 打開運行—-輸入命令“cmd”—-輸入命令“pip install tensorflow-cpu ”

GPU版本:

  1. 檢查GPU是否支持CUDN​:其實這一步是可以省略的,因爲CUDN支持的顯卡範圍特別廣,你基本只要確認你的GPU是不是NVIDIA的就可以了

  2. 下載CUDA:https://developer.nvidia.com/cuda-toolkit-archive,選擇CUDA Toolkit 8.0,再選Windows,下載完之後打開安裝,一路確認到底

  3. 下載cuDNN:https://developer.nvidia.com/rdp/cudnn-download,需要先註冊了才能下載,選擇Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0,下載後將其解壓縮就可以

  4. 設置環境變量:右鍵“我的電腦”,選擇“高級系統設置”,進入“環境變量”,進入“Path”
    添加”c:\cuda\bin”與”C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin”後點擊確認

  5. 打開運行—-輸入命令“cmd”—-輸入命令“pip install tensorflow-gpu ”,默認安裝最新版,如果需要指定版本可以輸入命令:“pip install tensorflow-gpu ==版本號”

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章