1. 程式人生 > >tensorflow(gpu) win10安裝 1060顯示卡驅動

tensorflow(gpu) win10安裝 1060顯示卡驅動

CUDA9.0+cuDNNv7+WIN10+1060顯示卡

一.驅動檔案下載

1.上tensorflow官網。檢視本機硬體和系統支援的tensorflow版本,以及對應的cuda和cuddnn版本。

https://www.tensorflow.org/install/install_windows?hl=zh-cn

2.百度進CUDA網站。查詢相應的CUDA驅動。

開啟預設下載是9.2版本,如下:


手動搜尋9.0版本,嚴格按照 tensorflow官網推薦的版本。

建議下載local安裝包。

3.百度cudnn進網站。

同樣進行搜尋:


下載前需要先進行賬號註冊和登入才能下載,按提示進行操作,下載此版本。


我的下載好是這兩個檔案。
在這裡插入圖片描述

二. 安裝

1.安裝CUDA驅動

安裝前需要把之前NVIDIA的驅動完全解除安裝掉,才能安裝上cuda的驅動。
解除安裝方法:進入系統安全啟動模式,把C盤Program Files 和 Program Files(X86)的以NVIDIA開頭的資料夾全部刪掉。(這一步是關鍵,不然後面安裝cuda會失敗)
然後重啟,執行cuda驅動安裝檔案,按照提示一步步執行,直至結束。

2.解壓cuDNN檔案。

將解壓後的bin和include資料夾中的檔案分別複製替換掉CUDA安裝目錄下的bin和include資料夾中的檔案,到此CUDA安裝完成。

3.tensorflow安裝。

按照tensorflow官網提供的,建議使用Anaconda安裝。

file
按操作等待操作完成。
如果事先沒安裝anaconda,可以選擇清華安裝包(官網下載anaconda超慢),百度 anaconda清華,根據系統下載新版本:
https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
在這裡插入圖片描述

4.驗證。

在這裡插入圖片描述