Ubuntu18.04 CUDA与cudnn安装
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)址
該網(wǎng)址記載了最低版本要求,比如我的1070,使用于cuda6.1及以上版本。由于Ubunt,18.04 默認(rèn)gcc版本為7.3,可以直接使用。因此這里不錯(cuò)調(diào)整,如需調(diào)整請(qǐng)參考——linux下gcc、g++不同版本的安裝和切換
該網(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 ./deviceQuerycudnn的安裝
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)題。
- 上一篇: linux下gcc、g++不同版本的安装
- 下一篇: Ubuntu18.04 显卡驱动安装