日韩性视频-久久久蜜桃-www中文字幕-在线中文字幕av-亚洲欧美一区二区三区四区-撸久久-香蕉视频一区-久久无码精品丰满人妻-国产高潮av-激情福利社-日韩av网址大全-国产精品久久999-日本五十路在线-性欧美在线-久久99精品波多结衣一区-男女午夜免费视频-黑人极品ⅴideos精品欧美棵-人人妻人人澡人人爽精品欧美一区-日韩一区在线看-欧美a级在线免费观看

歡迎訪問(wèn) 生活随笔!

生活随笔

當(dāng)前位置: 首頁(yè) > 运维知识 > Ubuntu >内容正文

Ubuntu

Ubuntu18.04 CUDA与cudnn安装

發(fā)布時(shí)間:2025/3/21 Ubuntu 55 豆豆
生活随笔 收集整理的這篇文章主要介紹了 Ubuntu18.04 CUDA与cudnn安装 小編覺(jué)得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.

cuda版本的選擇

CUDA與cudnn有多種安裝方式,目前搜索cuda下載默認(rèn)打開(kāi)的是10.2:https://developer.nvidia.com/cuda-downloads

進(jìn)入連接,按照要求選擇。本人使用的是1070,Ubuntu18.04,驅(qū)動(dòng)為440.59,因此這里直接選擇了10.2,如果你不是請(qǐng)參開(kāi)該網(wǎng)址以下兩個(gè)網(wǎng)址

  • https://developer.nvidia.com/cuda-gpus
  • 該網(wǎng)址記載了最低版本要求,比如我的1070,使用于cuda6.1及以上版本。由于Ubunt,18.04 默認(rèn)gcc版本為7.3,可以直接使用。因此這里不錯(cuò)調(diào)整,如需調(diào)整請(qǐng)參考——linux下gcc、g++不同版本的安裝和切換

  • https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
  • 該網(wǎng)址則展示了cuda于derivers之間的關(guān)系。

    cuda的安裝

    而cuda安裝包本身則有多種,因?yàn)槲覀冊(cè)谥袊?guó),考慮到服務(wù)器的問(wèn)題。建議選擇run file或者deb安裝。本人選擇的是run file。不過(guò)差別不大。

    在選擇后,下方會(huì)出現(xiàn)安裝時(shí)使用的命令。直接復(fù)制命令,下載完成后,輸入命令運(yùn)行即可,非常方便。

    注意如果已經(jīng)安裝驅(qū)動(dòng),則不需要再進(jìn)行安裝。其他都可以選擇y,安裝的cuda編程的例子可以用來(lái)做安裝驗(yàn)證,方法見(jiàn)最后。如果cuda安裝失敗,則考慮

    ubuntu-drivers devices #查看自己的顯卡及可以安裝的驅(qū)動(dòng)版本

    如果顯示目前使用的顯卡為集顯,則需要考慮顯卡的切換問(wèn)題,這個(gè)請(qǐng)參考該文章——

    Ubuntu 18.04 獨(dú)顯和集顯切換。

    無(wú)論你選擇的什么安裝方式,只要不選擇顯卡驅(qū)動(dòng)安裝,然后直接一路安裝即可。

    CUDA驗(yàn)證

    cd /home/fonttian/NVIDIA_CUDA-10.2_Samples/ sudo make cd /bin/x86_64/linux/release ./deviceQuery

    cudnn的安裝

    cudnn 的下載地址為——https://developer.nvidia.com/cudnn,不過(guò)依舊是需要在登錄后下載。

    登陸后,會(huì)出現(xiàn)列表以供選擇。如果選擇deb包,下載后直接sudo dpkg -i filename.deb即可。

    如果下載的是tar文件,則需要解壓縮并復(fù)制到對(duì)應(yīng)的文件夾。之后修改一下權(quán)限即可。

    sudo cp cuda/include/cudnn.h /usr/local/cuda/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

    cudnn檢查

    nvidia-smi # 查看nvidia信息,會(huì)顯示cuda版本號(hào) cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 # 查看cudnn版本

    如果是deb安裝的cudnn,會(huì)在/usr/src/cudnn_samples_v7有一些cudnn的例子。我們可以通過(guò)編譯mnistCUDNN sample進(jìn)行驗(yàn)證。

    cp -r /usr/src/cudnn_samples_v7/ $HOME cd $HOME/cudnn_samples_v7/mnistCUDNN make clean && make ./mnistCUDNN

    如果安裝成功將會(huì)有如下圖所示的:Test passed!

    (由于本人使用的是tar,因此網(wǎng)上搜了張圖片,該圖片來(lái)自于——ubuntu16.04安裝cuDNN的兩種方式以及驗(yàn)證)

    總結(jié)

    以上是生活随笔為你收集整理的Ubuntu18.04 CUDA与cudnn安装的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。

    如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò),歡迎將生活随笔推薦給好友。