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