[機器學習 ML] Windows 10, Python 3.6安裝Tensorflow-gpu

Source: Deep Learning on Medium


Go to the profile of 徐靖憲

這是我的第一篇關於機器學習的文章,希望大家可以多多指教,一起交流

以下是我電腦配備: 作業系統windows 10 64bits, python 3.6 , 顯卡1050ti


我以最簡單的指令來教大家如何快速

第一請先確認你的顯卡是否支援CUDA

第二從nvidia 官網中 https://www.nvidia.com/Download/index.aspx?lang=en-us 將驅動程式更新到最新


再來開始安裝吧!

(1) 安裝Anaconda 64bits版本 https://www.anaconda.com/download/

安裝完Anaconda後 在程式集找到Anaconda Prompt

先創建python3.5的環境( python 3.6尚未支援tensorflow)

conda create -n tensorflow-gpu pip python=3.5

(2)安裝完之後依然在Anaconda Prompt上激活tensorflow-gpu

activate tensorflow-gpu                 

(3)再來就是將tensorflow-gpu安裝上

pip install --ignore-installed --upgrade tensorflow-gpu

安裝CUDA 9.1 和 cuDNN 7.1

請到Nvidia CUDA官網 https://developer.nvidia.com/cuda-downloads 點選legacy releases 點選下載CUDA 9.1

點進之後 勾選下方圖片選擇的選項

下載Base installer

下載完後請在windows搜尋環境變數

確認一下是否有CUDA_PATH

接下來我們回到Anaconda Prompt輸入

nvcc -V

如果成功了話會出現

恭喜! 你的CUDA安裝成功

再來是安裝cuDNN 7.1

這是官網 https://developer.nvidia.com/rdp/cudnn-download 進去申請一個帳號就可以下載

選擇 Archived cuDNN releases

找到對應版本

下載完並且解壓縮,將檔安打開會發現有三個檔案,將他們複製。

複製到剛剛CUDA的FILE裡

最後再去環境變數增加兩個路徑

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1\libnvvp

大功告成! 接下來就看我們安裝的tensorflow-gpu可以跑嗎

進入tensorflow-gpu環境,輸入以下。如果沒有出現fail or mistakes就表示ok了

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

大家或多或許有被這超級麻煩的tensorflow-gpu安裝困擾到。我大概也是慢慢摸索一個月也才裝成功。過程是真的很煎熬阿。希望此文章有幫到各位。之後也會陸陸續續分享有關機器學習的文章。請大家多多指教並且一起討論

如果有問題,歡迎在下方留言並且幫我按個讚,你的讚是我寫文章的動力!