最近幾天做視頻數(shù)據(jù)集的時候要用到tensorflow2的gpu訓(xùn)練版本，要安裝對應(yīng)版本的cuda和cudnn堰乔，網(wǎng)上的教程五花八門，我嘗試了很多論壇的方法，有些能用有些不可以（也可能是我操作錯誤）斟赚，經(jīng)過兩天調(diào)試終于能夠運行了。
1介紹
使用的版本：Win10差油，python3.7汁展，tensorflow2.1-GPU，CUDA10.1, CUDNN7.6.5厌殉，顯卡驅(qū)動版本是425.31（用的IDE是pycharm）食绿。版本對應(yīng)關(guān)系主要可以參考?TF,CUDA,CUDNN版本對應(yīng)
注意：下載的版本都要一一對應(yīng)，錯了的話是用不了的公罕。
2NVIDIA驅(qū)動版本
先看看當前的驅(qū)動版本器紧，一般在電腦的右下角有個英偉達的標志，如果沒有那可能隱藏了楼眷，可以設(shè)置（具體請百度）
驅(qū)動版本不對應(yīng)的需要更新過程如下：
1先進入英偉達官網(wǎng)? ? ?英偉達官網(wǎng)
2在官網(wǎng)的上面有驅(qū)動程序，下拉有GEFORCE驅(qū)動程序罐柳，點這個
3點進去后下拉可以看到手動搜索的框掌腰，按照里面的設(shè)置找到電腦上的GPU型號，然后開始搜索
4拉到下面可以看到最新的版本张吉，下載下來后一路下一步傻瓜式安裝齿梁，最后可以看到前面的版本已經(jīng)變成了安裝的版本
2CUDA安裝
1下載CUDA? ?CUDA10.1? ?選擇win10的local部分，大概2.5G大小 其他版本的CUDA可以看這個NVIDIA CUDA各版本下載鏈接
下載慢的話把鏈接拷貝到迅雷里面肮蛹，飛速提升勺择。
2打開安裝第一個彈出的目錄是解壓的地方，安裝完成后會自動刪除伦忠，所以隨便找個地方就好了省核。
然后安裝目錄也選擇默認的，會放在這個目錄
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1?
3如何查看版本昆码，只要在命令行里進入安裝位置"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin"气忠，輸入 nvcc -V （后面那個V是大寫的）就可以看到10.1版本安裝好啦
在path里進行設(shè)置后可以直接輸入nvcc -V來查看結(jié)果，具體請看??CUDA在cmd中測試
3CUDNN安裝
1進入cudnn網(wǎng)站?cudnn下載赋咽，這個下載需要注冊賬號
2然后進入下載界面旧噪，根據(jù)前面的版本，我這里選的是10.1版本冬耿，也可以用迅雷下載 大概250M
3下載下來后解壓出來一個CUDA文件夾舌菜，里面有三個文件+一個txt
4把該文件里三個文件夾的內(nèi)容分別復(fù)制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1 這個文件夾對應(yīng)的下面，
然后再把cudnn壓縮包解壓出來的文件夾也放到這個里面亦镶，我重命名為cudnn如下圖第二個文件夾（這個文件夾就是cudnn解壓縮出來的日月，我只是改了個名字）
5path設(shè)置袱瓮，首先右鍵我的電腦屬性--左邊高級系統(tǒng)設(shè)置--高級--環(huán)境變量--系統(tǒng)變量--Path
編輯這個Path，然后新建爱咬，加入四個新path如下下圖（注意順序）尺借，然后cudnn就安裝完成了
6測試，在命令行進入到如下路徑C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite精拟，然后分別執(zhí)行bandwidthTest.exe和deviceQuery.exe會得到下面兩個RESULT=pass
4注意
1前面的做法都是親測可用的（至少我經(jīng)過測試可以使用）
2tensorflow2.1要求Microsoft VisualC2019版本（2017和2015都沒用）安裝軟件應(yīng)該是VC_redist.x64.exe這個可以自行百度下載燎斩。
5現(xiàn)在可以在使用tensorflow2.1快樂用GPU訓(xùn)練了
