国产 无码 综合区,色欲AV无码国产永久播放,无码天堂亚洲国产AV,国产日韩欧美女同一区二区

Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch

這篇具有很好參考價值的文章主要介紹了Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch。希望對大家有所幫助。如果存在錯誤或未考慮完全的地方,請大家不吝賜教,您也可以點(diǎn)擊"舉報違法"按鈕提交疑問。

一. 顯卡驅(qū)動、CUDA 、cuDNN 和 cuda版本的pytorch的關(guān)系

NVIDIA的顯卡在有驅(qū)動的前提下我們才能夠使用的。平時所說的顯卡的驅(qū)動和CUDA驅(qū)動不是一個東西,對于沒有用過顯卡的我來說開始就搞混了。CUDA 是用于顯卡并行計算的預(yù)算平臺,cuDNN專門加速深度學(xué)習(xí)算法在顯卡上的計算,個人理解相當(dāng)于加速器。cuda 版本的pytorch相當(dāng)于能夠在pytorch框架中很好的調(diào)用顯卡,用來處理算法。
該篇文章用于自己初次配置環(huán)境記錄。有些地方可能寫的不夠清晰,如有錯誤后面繼續(xù)更新。
cuda: Compute Unified Device Architecture,是一種有NVIDIA推出的通用并行計算架構(gòu), 該架構(gòu)使GPU能夠解決復(fù)雜的計算問題。

cudnn: 是NVIDIA 推出的用于深度神經(jīng)網(wǎng)絡(luò)的GPU加速庫,他強(qiáng)調(diào)性能,易用性和低內(nèi)存開銷。

cuda和cudnn的關(guān)系: cudnn是基于cuda架構(gòu)開發(fā)的專門用于深度神經(jīng)網(wǎng)絡(luò)的GPU加速庫。cuda可以理解為一個大的商圈,但這個商圈是空的,還未裝修。cudnn可以理解為裝修后的房間,例如負(fù)一樓專門針對游樂(深度神經(jīng)網(wǎng)絡(luò))裝修成大型游樂廠。

二. NVIDIA(英偉達(dá))顯卡驅(qū)動安裝

注意!ubuntu系統(tǒng)安裝一般會自帶驅(qū)動,在安裝NVIDIA驅(qū)動以前需要禁止系統(tǒng)自帶顯卡驅(qū)動nouveau:可以先通過指令lsmod | grep nouveau查看nouveau驅(qū)動的啟用情況,如果有輸出表示nouveau驅(qū)動正在工作,如果沒有內(nèi)容輸出則表示已經(jīng)禁用了nouveau。
圖片:
我的電腦有有輸出,表示nouveau啟動了,下面進(jìn)行nouveau的禁用:

  • 打開終端,在終端輸入下面指令

    sudo gedit /etc/modprobe.d/blacklist.conf
    

    之后彈出blacklist.conf文件:
    圖片:

  • 在blacklist.conf文件末尾加上這兩行,并保存:

    blacklist nouveau
    options nouveau modeset=0
    
  • 然后在終端輸入

    sudo update-initramfs -u  #應(yīng)用更改
    
  • 重啟,就禁止了ubuntu20.04自帶的nouveau顯卡驅(qū)動了,接下來我們就可以安心的安裝自己想要的NVIDIA的驅(qū)動程序了

三. 安裝顯卡驅(qū)動

ubuntu20.04 安裝NVIDIA驅(qū)動很容易,只只需要打開系統(tǒng)設(shè)置->軟件和更新->附加驅(qū)動->選擇NVIDIA驅(qū)動->應(yīng)用更改(我這里選擇的是510版本的驅(qū)動)。該界面會自動根據(jù)電腦上的GPU顯示推薦的NVIDIA顯卡驅(qū)動。
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch
稍等即可完成。

安裝cuda和cudnn前的準(zhǔn)備工作

安裝一系列的版本的查詢

到pytorch官網(wǎng)查詢,能夠支持的CUDA的版本有哪些,最好是對應(yīng)的版本,然后在到cudnn的官網(wǎng)查詢cudnn支持的CUDA的版本,

四. 安裝CUDA與測試

4.1下載與安裝

NVIDIA(英偉達(dá))顯卡驅(qū)動安裝完成后,重啟電腦,在終端輸入nvidia-smi輸出如下圖所示的結(jié)果就表示安裝成功了。下圖中左邊的紅色框內(nèi)表明,剛剛安裝好的顯卡驅(qū)動版本(我的510版本),右面的紅色框顯示的是這個版本的顯卡驅(qū)動支持的CUDA最高版本為11.6(注意這里并不表明已經(jīng)安裝了CUDA,而是剛安裝的顯卡驅(qū)動支持的CUDA的最高版本)
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch
知道自己要安裝的CUDA版本后,到NVIDIA官網(wǎng)的CUDA下載頁面(點(diǎn)擊這個藍(lán)色字體即可)找到對應(yīng)的CUDA版本,如圖所示,我選擇的是CUDA 11.6的版本。
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch
點(diǎn)進(jìn)去之后會出現(xiàn)如下圖所示的界面,依次選擇1234。然后彈出三種安裝方法,根據(jù)安裝經(jīng)驗(yàn)這里推薦采用 5 runfile(local)方法,即選擇如下圖中的紅圈中的選項(xiàng)。這是由于CUDA的安裝過程需要很多依賴庫文件,CUDA的run文件雖然比另外兩種安裝方法的文件大,但是它包含了所有的依賴庫文件,所以采用相對來說很容易安裝成功。
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch
在安裝CUDA11.0之前需要首先安裝一些相互依賴的庫文件,在終端運(yùn)行下面指令:

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

然后在返回上面界面,依次復(fù)制下面紅框中兩條指令到終端中。(解釋一下weget:下載安裝包到終端運(yùn)行的目錄中,所以下面直接sudo運(yùn)行即可。)
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch
運(yùn)行上面指令后,會彈出如下界面,按?按鍵,選擇Continue 在按下回車,然后再輸入accept
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch接著,如下圖所示,在彈出的界面中通過Enter鍵,取消Driver和450.51.05的安裝(這里不取消會覆蓋掉上面安裝的驅(qū)動,會出問題),然后點(diǎn)擊Install,等待…
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch

4.2配置CUDA環(huán)境變量

CUDA安裝完成后,需要配置變量環(huán)境才能正常使用。首先在終端輸入

sudo gedit ~/.bashrc

打開如下圖所示的.bashrc文件,然后,如下圖所示在.bashrc文件的最后添加以下CUDA環(huán)境變量配置信息(cuda-11.6 是版本信息,不同版本需要改到你自己的版本號),然后保存文件。

export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch

4.3 CUDA測試

最后,在終端輸入source ~/.bashrc或者重新啟終端使之生效。這時,我們就可以在終端輸入nvcc -V查看CUDA的安裝信息,如下圖所示,至此CUDA安裝成功。
其中,LD_LIBARARY_PATH為動態(tài)庫搜索路徑。
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch
以上的添加環(huán)境變量官網(wǎng)有教程,也可以參照,我目前沒找到在哪里找到更新。

五. 安裝cuDNN

首先進(jìn)入cuDNN官網(wǎng)
網(wǎng)址:https://developer.nvidia.com/rdp/cudnn-archive
先注冊個nvidia賬號,用賬號登才能下載。。。
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch
選擇對應(yīng)版本的cuDNN,因?yàn)槲业腃UDA 是11.6版本的,所以我選擇的是上圖紅色框中的,使用11.x版本。點(diǎn)開。
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch
我這里選擇的是紅框中的,感覺過程比較簡單,其他的目前沒嘗試過。下面的額以一個紅框是我的下載目錄,一般的下載目錄就是主目錄下的名為‘下載’的文件夾。因?yàn)榘惭b包是xz結(jié)尾的壓縮文件,先用xz -d cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive.tar.xz,在用tar xf cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive.tar 解壓,加壓之后就是第二個紅框里面的。
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch
接下來就是安裝:這里參考的是官方文檔

  • 鼠標(biāo)右鍵 點(diǎn)擊在終端打開,進(jìn)入終端。
  • 依次運(yùn)行指令
    sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
    sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
    sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
    
  • 將所有的.h 頭文件都復(fù)制到 /usr/local/cuda 中,完成。

cudnn 測試

在終端中輸入如下指令:

cat /usr/local/cuda-11.6/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

輸出紅色框里面的信息就是成功。
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch
笨方法:也可以直接去目錄下看看這個cudnn_version.h文件,可以點(diǎn)開紅框內(nèi)容看看 。其中也會有右面終端中輸出的信息。
Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch

五 安裝pytorch

首先是安裝pycharm和anaconda:
-----首先是安裝pycharm:參考link
-----安裝anaconda:參考link
官網(wǎng)找到合適的pycharm版本:創(chuàng)建環(huán)境,直接用命令行安裝。文章來源地址http://www.zghlxwxcb.cn/news/detail-473884.html

到了這里,關(guān)于Ubuntu20.04安裝CUDA cuDNN 以及對應(yīng)的pytorch的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!

本文來自互聯(lián)網(wǎng)用戶投稿,該文觀點(diǎn)僅代表作者本人,不代表本站立場。本站僅提供信息存儲空間服務(wù),不擁有所有權(quán),不承擔(dān)相關(guān)法律責(zé)任。如若轉(zhuǎn)載,請注明出處: 如若內(nèi)容造成侵權(quán)/違法違規(guī)/事實(shí)不符,請點(diǎn)擊違法舉報進(jìn)行投訴反饋,一經(jīng)查實(shí),立即刪除!

領(lǐng)支付寶紅包贊助服務(wù)器費(fèi)用

相關(guān)文章

  • Ubuntu 20.04安裝CUDA & CUDNN 手把手帶你擼

    Ubuntu 20.04安裝CUDA & CUDNN 手把手帶你擼

    之前一直使用CPU做訓(xùn)練,最近手上多了臺單卡1080Ti顯卡主機(jī),于是開始研究GPU訓(xùn)練。用GPU訓(xùn)練一定會使用CUDA了,剛開始接觸CUDA非常非常頭痛,對小白很不友好,我?guī)缀跽苏惶?,中途想放棄過很多次,最后終于弄好了,回頭看看其實(shí)不是很難,但是網(wǎng)上文章又多又雜,

    2024年02月02日
    瀏覽(25)
  • 筆記--Ubuntu20.04安裝Nvidia驅(qū)動、CUDA Toolkit和CUDA CuDNN

    筆記--Ubuntu20.04安裝Nvidia驅(qū)動、CUDA Toolkit和CUDA CuDNN

    目錄 1--安裝Nvidia驅(qū)動 2--安裝CUDA 2-1--禁用nouveau 2-2--選擇CUDA Toolkit 2-3--下載和安裝CUDA Toolkit 2-4--配置環(huán)境變量 2-5--測試是否安裝成功: 3--安裝CUDA CuDNN 4--測試pytorch能否使用Cuda ① 查看可安裝的Nvidia驅(qū)動版本: ② 安裝相應(yīng)版本的Nvidia驅(qū)動: 博主這里選擇的是第一個,也可以安

    2024年02月02日
    瀏覽(24)
  • 【ubuntu環(huán)境配置】超詳細(xì)ubuntu20.04/22.04安裝nvidia驅(qū)動/CUDA/cudnn

    【ubuntu環(huán)境配置】超詳細(xì)ubuntu20.04/22.04安裝nvidia驅(qū)動/CUDA/cudnn

    nvidia顯卡驅(qū)動安裝方式有三種:使用ubuntu附加驅(qū)動的方式;使用命令行方式安裝;使用.run文件的方式進(jìn)行安裝, 點(diǎn)擊菜單中的Additional Drivers選擇適合的驅(qū)動版本進(jìn)行安裝,該方法最方便快捷(但有時會翻車) 更新所有的軟件包 安裝顯卡驅(qū)動 詳見我的另一篇博客Ubunut20.04/2

    2024年01月22日
    瀏覽(35)
  • 【Ubuntu 20.04LTS系統(tǒng)】安裝CUDA11.8、cuDNN,可進(jìn)行CUDA版本切換

    【Ubuntu 20.04LTS系統(tǒng)】安裝CUDA11.8、cuDNN,可進(jìn)行CUDA版本切換

    https://mirrors.tuna.tsinghua.edu.cn/help/ubuntu/ 更新軟件列表和必要的依賴項(xiàng) 步驟一: 下載CUDA安裝包 進(jìn)行CUDA和cuDNN的選擇,也可以直接根據(jù)官方推薦進(jìn)行下載安裝。 從Nvidia官網(wǎng)下載CUDA https://developer.nvidia.com/cuda-downloads 下方鏈接,選擇更多版本 https://developer.nvidia.com/cuda-toolkit-archiv

    2024年02月16日
    瀏覽(61)
  • ubuntu20.04系統(tǒng)4060安裝cuda11.8和cudnn8.6

    ubuntu20.04系統(tǒng)4060安裝cuda11.8和cudnn8.6

    在終端輸入: 安裝代碼: 重啟電腦(這一步很重要,不重啟沒有效果): 安裝命令 CUDA默認(rèn)安裝在/usr/local/目錄下,一般Change Toolkit Install Path可以不做修改。但如果是普通用戶安裝,需設(shè)定安裝路徑為用戶主目錄下,光標(biāo)移動到Change Toolkit Install Path按回車,手動修改安裝路徑后按

    2024年01月16日
    瀏覽(29)
  • Ubuntu 20.04 安裝NVIDIA顯卡驅(qū)動+cuda 11.7+cudnn 8.4

    Ubuntu 20.04 安裝NVIDIA顯卡驅(qū)動+cuda 11.7+cudnn 8.4

    參考: https://zhuanlan.zhihu.com/p/59618999 https://blog.csdn.net/linhai1028/article/details/79445722/ Windows+Ubuntu從雙系統(tǒng)安裝到CUDA cuDNN docker 配置K21 https://blog.csdn.net/qq_45831128/article/details/127060475 https://blog.csdn.net/kunhe0512/article/details/125061911 Ubuntu 20.04 英偉達(dá) RTX 3050 Ti 顯卡 1)Ubuntu下查看Nvidia顯卡的

    2023年04月10日
    瀏覽(32)
  • ubuntu20.04顯卡驅(qū)動cuda cudnn conda TensorRT安裝及配置

    ubuntu20.04顯卡驅(qū)動cuda cudnn conda TensorRT安裝及配置

    如果要使用 TensorRT ,請注意CUDA 支持的最高算力要大于等于GPU本身的算力, 可以先看3小節(jié) conda和pip換源直接看2.3小節(jié) 本人已在ubuntu20.04下安裝成功。其他版本步驟應(yīng)該差不多 如果帖子有幫助,感謝一鍵三連, ^_^ 部署有問題的小伙伴歡迎留言和加 Q 裙- 472648720 BEV各算法環(huán)境部

    2024年01月22日
    瀏覽(49)
  • Ubuntu 20.04 LTS 系統(tǒng)下 安裝Nvidia 顯卡驅(qū)動、CUDA、cuDNN, 并可進(jìn)行CUDA版本切換

    Ubuntu 20.04 LTS 系統(tǒng)下 安裝Nvidia 顯卡驅(qū)動、CUDA、cuDNN, 并可進(jìn)行CUDA版本切換

    因?yàn)樽錾疃葘W(xué)習(xí)的研究項(xiàng)目,為全新機(jī)器在Ubuntu 20.04 LTS 系統(tǒng)下 安裝Nvidia 顯卡驅(qū)動、Cuda、Cudnn。并進(jìn)行CUDA版本切換 成功安裝完成了,寫個記錄。 步驟一: 安裝更新軟件列表和依賴項(xiàng) 在安裝Nvidia顯卡驅(qū)動之前,需要更新軟件列表和必要的依賴項(xiàng)。 步驟二: 查看GPU型號,并

    2024年02月13日
    瀏覽(28)
  • CUDA11.1、cuDNN8.6.0、Tensorrt8.5.3,ubuntu20.04安裝過程記錄

    CUDA11.1、cuDNN8.6.0、Tensorrt8.5.3,ubuntu20.04安裝過程記錄

    CUD11.1 下載地址:CUDA Toolkit Archive | NVIDIA Developer ?安裝: 對于不是sudo用戶,可以不執(zhí)行sudo,不過沒辦法裝到/usr/local/,可以裝到你有權(quán)限的文件夾目錄。 裝完后,需要增加環(huán)境路徑到~/.bashrc并source ?記得source ~/.bashrc使得環(huán)境生效 cuDNN8.6.0,這個版本我是根據(jù)我需要安裝的T

    2024年02月15日
    瀏覽(59)
  • ubuntu 20.04 環(huán)境下安裝CUDA 11.8, cuDNN v8.6.0和TensorRT 8.6.0(deb方式)

    ubuntu 20.04 環(huán)境下安裝CUDA 11.8, cuDNN v8.6.0和TensorRT 8.6.0(deb方式) 取消勾選驅(qū)動, 下一步 添加環(huán)境變量 最后一行添加: 保存關(guān)閉后,應(yīng)用一下更改 下載 cuDNN v8.6.0 for CUDA 11.x 導(dǎo)入CUDA GPG key 刷新存儲庫元數(shù)據(jù) 進(jìn)入 cudnn-local 目錄 安裝 參考文檔: 1 NVIDIA CUDA Installation Guide for Linu

    2024年02月09日
    瀏覽(49)

覺得文章有用就打賞一下文章作者

支付寶掃一掃打賞

博客贊助

微信掃一掃打賞

請作者喝杯咖啡吧~博客贊助

支付寶掃一掃領(lǐng)取紅包,優(yōu)惠每天領(lǐng)

二維碼1

領(lǐng)取紅包

二維碼2

領(lǐng)紅包