為了方便組員操作,簡單寫一下CUDA的配置啦。
首先你需要一台電腦,有NVDIA顯卡的那種(就那個煤氣灶,你懂我意思吧),然后就繼續往下走吧,如果沒有的話可以找一下右上角的紅叉了,這篇文章對你可能么得用處了。
怎么知道我還要不要往下看呢,右鍵->我的電腦->管理;設備管理器->顯示適配器:
下面就會看到有NVIDA的顯卡啦,那有的話恭喜你,你不用關這篇文章了!你可以繼續往下看惹!
首先我們更新一下我們的顯卡驅動到最新版本,這里建議大家去NVIDA的官網更新,傳送門:https://www.geforce.cn/drivers,里面有三種操作,自動更新:GeForce Experience(這是最吼的,懶人首選),還有手動查找顯卡型號的驅動和自動檢測GPU的,然后大家就跟着一步步往下裝更新到最新啦~
Then,你需要一個VS,不過VS的安裝就相對親切友好了,需要注意的是安裝時記得勾選VC++,不然你大概率會遇到CUDA編譯錯誤。Finally,我們安裝一下Python,這里需要注意一下的是,如果VS2017安裝的時候,你已經裝了Python,就表再裝了,打開VS2017, 點擊Tools->Python->Python Environments,應該可以看到Python 3.6,在下面有個folder,大概是“c:\Program Files(x86)\Microsoft Visual Studio\Shared\Python36_64”,把這個字符串copy下來。然后打開Settings->Home->About->System info,在彈出的窗口中選擇Advanced system settings->Advanced->Environment Variables->System variables->Path->Edit->New,把剛才的python環境變量字符串paste進來。如果有多個python環境,建議把一些舊的版本卸載先,保證你的機器沒有那么多垃圾。
點擊OK后,再Win+R,輸入cmd打開命令提示符窗口,輸入Python,就可以正常使用了。
然后准備工作到這里就基本完成啦~進入正題:
1、安裝CUDA
首先我們又要去一個神奇的小網站,https://developer.nvidia.com/cuda-toolkit-archive,你會看到一些神奇的版本
比如這種,但是千萬不要就這么可愛的去下載最新版本,因為通常情況下,你的顯卡是玩不轉這種操作的,那么我們選什么呢,比如我的菜雞顯卡GeForce 940MX,就比較適合9.0版本的,那么我們就選9.0吧
點擊后咱這么配置,大家電腦大概都是Windows吧(是Linux的當我沒說2333),還是給出Linux(CentOS為例)和Windows(Win10)的選項選擇:
Tips:這里建議選local版本,如果你選network版本的話,下載到一半崩了就真的血崩了,畢竟local版本崩了還能再開,network崩了就是每次都要重新下1.4G。
打開安裝包,選一個安裝路徑,然后手動將CUDA庫添加到PATH里面去。
這里有個小坑,在安裝時可能你要看一下VS的問題,有的時候它跟VS Integration兼容會出現一些問題,如果出現問題請不要勾選VS那項,安裝完成后,找到CUDAVisualStudioIntegration\extras\visual_studio_integration\MSBuildExtensions,復制下面的所有文件,復制到自己VS2017的安裝目錄下面: C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\Common7\IDE\VC\VCTargets\BuildCustomizations,然后就可以愉快使用了。
如果你希望能讓CUDA在VS當中新建項目並且愉快使用的話,知乎小姐姐是這么說的(大概是小姐姐)。
2、安裝cuDNN
Tips:最麻煩的地方開始了,前方高能!還是以Win10為例
還是需要個神奇的小網站:https://developer.nvidia.com/rdp/cudnn-archive,我們從里面找到適合我們的cuDNN版本,
cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0 cuDNN v7.0.5 Library for Windows 10
(小聲bibi:其實Ubuntu16.04的也在這里了,一樣的操作)
然鵝,當你滿心歡喜打開下載鏈接的時候,等待你的是……
等會等會,Membership Required?我用我畢生的英語水平(微軟翻譯),好像意思是要注冊成會員?歐……↑上面就是本教程里面最麻煩的一步了,在下載cuDNN之前需要注冊Nvidia會員並驗證郵箱。比較騷一點的可以選擇微信登錄,能省則省。然后在你基本窒息的時候你就成功get到了 cudnn64_7.dll 這個文件,我們愉快的把這個文件復制到CUDA的bin目錄下即可,就是你之前安裝CUDA那個,大概長相是\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin。
到這里基本上CUDA和cuDNN就配好惹,驚不驚喜意不意外?
跟我默念:刺蝟最可愛!