爲了方便組員操做,簡單寫一下CUDA的配置啦。python
首先你須要一臺電腦,有NVDIA顯卡的那種(就那個煤氣竈,你懂我意思吧),而後就繼續往下走吧,若是沒有的話能夠找一下右上角的紅叉了,這篇文章對你可能麼得用處了。微信
怎麼知道我還要不要往下看呢,右鍵->個人電腦->管理;設備管理器->顯示適配器:網站
下面就會看到有NVIDA的顯卡啦,那有的話恭喜你,你不用關這篇文章了!你能夠繼續往下看惹!ui
首先咱們更新一下咱們的顯卡驅動到最新版本,這裏建議你們去NVIDA的官網更新,傳送門:https://www.geforce.cn/drivers,裏面有三種操做,自動更新:GeForce Experience(這是最吼的,懶人首選),還有手動查找顯卡型號的驅動和自動檢測GPU的,而後你們就跟着一步步往下裝更新到最新啦~spa
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,就能夠正常使用了。翻譯
而後準備工做到這裏就基本完成啦~進入正題:code
首先咱們又要去一個神奇的小網站,https://developer.nvidia.com/cuda-toolkit-archive,你會看到一些神奇的版本blog
好比這種,可是千萬不要就這麼可愛的去下載最新版本,由於一般狀況下,你的顯卡是玩不轉這種操做的,那麼咱們選什麼呢,好比個人菜雞顯卡GeForce 940MX,就比較適合9.0版本的,那麼咱們就選9.0吧教程
點擊後咱這麼配置,你們電腦大概都是Windows吧(是Linux的當我沒說2333),仍是給出Linux(CentOS爲例)和Windows(Win10)的選項選擇:ip
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當中新建項目而且愉快使用的話,知乎小姐姐是這麼說的(大概是小姐姐)。
二、安裝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就配好惹,驚不驚喜意不意外?
跟我默唸:刺蝟最可愛!