通俗的講解深度學(xué)習(xí)中CUDA,cudatookit,cudnn和pytorch的關(guān)系
CUDA
CUDA是顯卡廠商N(yùn)VIDIA推出的運(yùn)算平臺(tái)。CUDA™是一種由NVIDIA推出的通用并行計(jì)算架構(gòu)，是一種并行計(jì)算平臺(tái)和編程模型，該架構(gòu)使GPU能夠解決復(fù)雜的計(jì)算問題。CUDA英文全稱是Compute Unified Device Architecture。
我們可以通過nvidia-smi命令查看cuda版本號(hào)。
如上圖，我的電腦cuda版本是11.7
CUDA Toolkit
CUDA Toolkit可以理解成一個(gè)工具包，主要包含了CUDA-C和CUDA-C++編譯器、一些科學(xué)庫和實(shí)用程序庫、CUDA和library API的代碼示例、和一些CUDA開發(fā)工具。
cudatookit版本有時(shí)會(huì)被簡稱為cuda版本，這也是我們經(jīng)常搞混的一個(gè)原因。
cuDNN
cuDNN的全稱為NVIDIA CUDA® Deep Neural Network library，是NVIDIA專門針對深度神經(jīng)網(wǎng)絡(luò)中的基礎(chǔ)操作而設(shè)計(jì)基于GPU的加速庫。cuDNN為深度神經(jīng)網(wǎng)絡(luò)中的標(biāo)準(zhǔn)流程提供了高度優(yōu)化的實(shí)現(xiàn)方式。
簡單來說，cuDNN是基于CUDA的深度學(xué)習(xí)GPU加速庫，有了它才能在GPU上完成深度學(xué)習(xí)的計(jì)算。
Pytorch
pytorch是基于CUDA的深度學(xué)習(xí)框架，因此，pytorch的版本必須依賴于cuda toolkit的版本。
如果你在讀上面的一些名詞的時(shí)候感覺模糊不清，那么可以直接來看下面的總結(jié)。（當(dāng)然還是建議把不懂的地方搜索明白）
CUDA看作是一個(gè)工作臺(tái)，上面配有很多工具，如錘子、螺絲刀等。cuDNN是基于CUDA的深度學(xué)習(xí)GPU加速庫，有了它才能在GPU上完成深度學(xué)習(xí)的計(jì)算。它就相當(dāng)于工作的工具，比如它就是個(gè)扳手。但是CUDA這個(gè)工作臺(tái)買來的時(shí)候，并沒有送扳手。想要在CUDA上運(yùn)行深度神經(jīng)網(wǎng)絡(luò)，就要安裝cuDNN，就像你想要擰個(gè)螺帽就要把扳手買回來。這樣才能使GPU進(jìn)行深度神經(jīng)網(wǎng)絡(luò)的工作，工作速度相較CPU快很多。
總結(jié)一下，食物鏈頂端的是CUDA這個(gè)工作臺(tái)，也就是我們電腦的硬件配置，我們要先查看它的版本，然后根據(jù)這個(gè)來看我們都能下載多高的cudatookit版本，然后根據(jù)cudatookit版本來選擇cudnn版本和可支持的pytorch版本
因此配置環(huán)境的流程為：
查看CUDA版本 ——> 選擇cudatookit版本 ——> 選擇cudnn版本 + pytorch版本
到此這篇關(guān)于通俗的講解深度學(xué)習(xí)中CUDA,cudatookit,cudnn和pytorch的關(guān)系的文章就介紹到這了,更多相關(guān)CUDA,cudatookit,cudnn和pytorch的關(guān)系內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Python中集成了專用于處理csv文件的庫，名為：csv，本文給大家介紹了Python使用csv庫處理CSV文件的方法及csv庫中4個(gè)常用的對象，結(jié)合實(shí)例代碼給大家介紹的非常詳細(xì),需要的朋友可以參考下2023-06-06
win10下安裝Anaconda的教程（python環(huán)境+jupyter_notebook)
Anaconda指的是一個(gè)開源的Python發(fā)行版本，其包含了conda、Python等180多個(gè)科學(xué)包及其依賴項(xiàng)。這篇文章主要介紹了win10下安裝Anaconda（python環(huán)境+jupyter_notebook),需要的朋友可以參考下2019-10-10
深入了解python基于tkinter寫的畫圖項(xiàng)目
這篇文章主要為大家介紹了python基于tkinter寫的畫圖項(xiàng)目，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2021-12-12
Python文件讀寫處理日常任務(wù)終極工具實(shí)例
Python文件的讀寫操作時(shí),有很多需要考慮的細(xì)節(jié),這包括文件打開方式、讀取和寫入數(shù)據(jù)的方法、異常處理等,在本文中,將深入探討Python中的文件操作,旨在提供全面的指南,幫你充分了解Python文件的讀寫2023-11-11
這篇文章主要介紹了Flask路由尾部加不加斜杠有什么區(qū)別，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-09-09
openCV中值濾波和均值濾波的代碼實(shí)現(xiàn)
在我們生活中的有很多時(shí)候都可以用到濾波，例如美顏的磨皮功能，本文就詳細(xì)的介紹了openCV中值濾波和均值濾波的代碼實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2022-03-03
Python+Pytorch實(shí)戰(zhàn)之彩色圖片識(shí)別
這篇文章主要為大家詳細(xì)介紹了如何利用Python+Pytorch實(shí)現(xiàn)彩色圖片識(shí)別功能，文中的示例代碼講解詳細(xì)，感興趣的小伙伴可以了解一下2022-09-09
圖神經(jīng)網(wǎng)絡(luò)GNN算法基本原理詳解
這篇文章主要為大家介紹了圖神經(jīng)網(wǎng)絡(luò)GNN算法基本原理詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-05-05
淺談Python實(shí)現(xiàn)貪心算法與活動(dòng)安排問題
本篇文章主要介紹了淺談Python實(shí)現(xiàn)貪心算法與活動(dòng)安排問題，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2017-12-12
這篇文章主要為大家詳細(xì)介紹如何通過Python分析股票的收益率，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2022-04-04