如何安裝CUDA、cuDNN 到Ubuntu 20.04? - 叩頂窩客

文章推薦指數: 80 %
投票人數:10人

玩深度學習第一件事,先把CUDA、cuDNN 裝好!先確認手上的GPU 是否有支援CUDA,緊接著下載CUDA 安裝,過程中他也會自動幫你安裝正確的驅動。

AI神經網路自走車•DeepLearning如何安裝CUDA、cuDNN到Ubuntu20.04? 2021-07-13byJNoComments玩深度學習第一件事,就是先把CUDA、cuDNN裝好!一聽到「裝環境」我們總是眉頭一皺,這過程肯定有許多坑等著。

但其實真的不難,就讓我們看下去!目錄1Step1:確認GPU支援的CUDA版本2Step2:安裝CUDA3Step3:安裝nvidia-cuda-toolkit4Step4:安裝cuDNNStep1:確認GPU支援的CUDA版本先確認手上的GPU是否有支援CUDA(如果不支援怎麼還買XD),如果已經安裝可以使用以下指令查詢顯卡規格sudolshw-numeric-Cdisplay以我為例,會列出一張GTX750(太窮先拿舊顯卡玩玩練習),以及IntelCPU的核顯*-display description:VGAcompatiblecontroller product:GM107[GeForceGTX750][10DE:1381] vendor:NVIDIACorporation[10DE] physicalid:0 businfo:[email protected]:01:00.0 version:a2 width:64bits clock:33MHz capabilities:pmmsipciexpressvga_controllerbus_mastercap_listrom configuration:driver=nvidialatency=0 resources:irq:36memory:f6000000-f6ffffffmemory:e0000000-efffffffmemory:f0000000-f1ffffffioport:e000(size=128)memory:f7000000-f707ffff *-display description:VGAcompatiblecontroller product:XeonE3-1200v3/4thGenCoreProcessorIntegratedGraphicsController[8086:412] vendor:IntelCorporation[8086] physicalid:2 businfo:[email protected]:00:02.0 version:06 width:64bits clock:33MHz capabilities:msipmvga_controllerbus_mastercap_listrom configuration:driver=i915latency=0 resources:irq:34memory:f7400000-f77fffffmemory:d0000000-dfffffffioport:f000(size=64)memory:c0000-dffff知道型號後到以下網址查詢是否支援:https://developer.nvidia.com/cuda-gpusStep2:安裝CUDA使用剛安裝好、乾淨的Ubuntu來裝比較不會出錯,如果已經安裝過舊版CUDA可能需要先移除。

安裝CUDA麻煩的地方就是他的版本有最低需求顯卡驅動版號,懶人法就是不要自己安裝驅動!先去下載CUDA安裝,過程中他就會自動幫你安裝正確的驅動,很人性!到nvidia官網下載CUDAhttps://developer.nvidia.com/cuda-toolkit-archive依照你的系統選擇,這邊以x86Ubuntu20.04為例,並且點選dev(local)的安裝方法。

下面就會列出安裝指令,依序把他複製貼上就完成了!安裝完後,先重開機!緊接著確認Driver是否也有在裝CUDA時正確裝上。

如果有回傳表示ok!$nvidia-smi +-----------------------------------------------------------------------------+ |NVIDIA-SMI470.42.01DriverVersion:470.42.01CUDAVersion:11.4| |-------------------------------+----------------------+----------------------+ |GPUNamePersistence-M|Bus-IdDisp.A|VolatileUncorr.ECC| |FanTempPerfPwr:Usage/Cap|Memory-Usage|GPU-UtilComputeM.| |||MIGM.| |===============================+======================+======================| |0NVIDIAGeForce...On|00000000:01:00.0Off|N/A| |31%47CP01W/38W|262MiB/981MiB|0%Default| |||N/A| +-------------------------------+----------------------+----------------------+Step3:安裝nvidia-cuda-toolkit這步驟可選,但可能以後會用到CUDA比較底層的功能,就順手安裝一下sudoaptinstallnvidia-cuda-toolkit安裝完後用以下指令確認版號$nvcc-V nvcc:NVIDIA(R)Cudacompilerdriver Copyright(c)2005-2019NVIDIACorporation BuiltonSun_Jul_28_19:07:16_PDT_2019 Cudacompilationtools,release10.1,V10.1.243Step4:安裝cuDNN一樣需要到nvidia官網下載,但必須先申請nvidiadeveloper的帳號才行下載。

https://developer.nvidia.com/rdp/cudnn-archivecuDNN也必須對應CUDA版號來選擇,這邊我們選forCUDA11.x的x86_64Linux版下載再依序使用以下指令安裝,或直接參考官方文件指令#解壓檔案,需替換成你下載的檔名 tar-xzvfcudnn-11.3-linux-x64-v8.2.1.32.tgz #將解壓的檔案搬到對應資料夾下就完成! sudocpcuda/include/cudnn*.h/usr/local/cuda/include sudocp-Pcuda/lib64/libcudnn*/usr/local/cuda/lib64 sudochmoda+r/usr/local/cuda/include/cudnn*.h/usr/local/cuda/lib64/libcudnn*完成!可以開始享受GPU的加速啦!如果覺得我文章內容對你有幫助的話,請在文章後面幫我按5個讚!讓我知道大家都喜歡什麼內容哦!參考資料Ubuntu16.04安裝CUDA、cuDNN教學https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#installlinuxHowtoinstallthelatestNvidiadriversonUbuntu16.04XenialXerusLinux如何用指令查詢Nvidia驅動程式版本與GPU顯示卡資訊? 相關文章AICUDADIYAuto-DriveCarUbuntu20.04 WrittenbyJ雖然大學唸的是生物,但持著興趣與熱情自學,畢業後轉戰硬體工程師,與宅宅工程師們一起過著沒日沒夜的生活,做著台灣最薄的intel筆電,要與macbookair比拼。

離開後,憑著一股傻勁與朋友創業,再度轉戰軟體工程師,一手扛起前後端、雙平台app開發,過程中雖跌跌撞撞,卻也累計不少經驗。

可惜不是那1%的成功人士,於是加入其他成功人士的新創公司,專職開發後端。

沒想到卻在採前人坑的過程中,拓寬了眼界,得到了深層的領悟。

«Previouspost巨型服務架構:Ch5輸入輸出設計閱讀筆記 Nextpost»如何在JetsonNano上安裝OpenCV?Aboutme J雖然大學唸的是生物,但持著興趣與熱情自學,畢業後轉戰硬體工程師,與宅宅工程師們一起過著沒日沒夜的生活。

之後憑著一股傻勁與朋友創業,再度轉戰軟體工程師,一手扛起前後端、雙平台app開發,過程中雖跌跌撞撞,卻也累計不少經驗。

可惜不是那1%的成功人士,於是加入其他成功人士的新創公司,專職開發後端。

沒想到卻在採前人坑的過程中,拓寬了眼界,得到了深層的領悟。

...moreSponserCategoriesAngular(2)Arduino(5)ESP32(2)RaspberryPiPico(3)AWS(13)Blockchain(1)DeepLearning(5)DevOps(8)Frontend(1)Golang(1)Kafka(2)Python(10)RaspberryPi(23)BananaPi(3)JetsonNano(2)OdroidHC4(3)RockPi(2)ROS(2)Wordpress(7)健身(1)創客計畫(14)AI神經網路自走車(8)智慧家居(6)說人話的名詞解釋(3)那些踩過的坑(6)開發哲學(2)閱讀筆記(2)SponserRelated 如何在JetsonNano上安裝TensorFlow以及PyTorch? 如何在JetsonNano上安裝OpenCV? 用Raspberrypipico做PWMMotorControl 用Raspberrypipico與HC-SR04做超音波距離測量 如何用Raspberrypipico做I2CSlave?使用MicroPython與Raspberrypi



請為這篇文章評分?