Ubuntu18.04安裝N卡驅動、Cuda10.0、cuDNN7.5.0

Ubuntu18.04安裝N卡驅動、CUDA10.0、cuDNN7.5.0

N卡驅動

先添加源:

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

然後執行:

ubuntu-drivers devices

查看合適的顯卡驅動,顯示如下:

== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001F11sv00001462sd0000126Abc03sc00i00
vendor   : NVIDIA Corporation
driver   : nvidia-driver-418 - third-party free
driver   : nvidia-driver-430 - third-party free recommended
driver   : xserver-xorg-video-nouveau - distro free builtin

這裏雖然推薦安裝nvidia-driver-430,但是親測重啓後無法進入圖形界面,因此選擇安裝nvidia-driver-418

sudo apt install nvidia-driver-418

安裝並重啓後,可以打開N卡設置界面,確認驅動安裝成功:

nvidia-settings 

在這裏插入圖片描述

CUDA 10.0

先從官網上下載安裝程序,本人選擇的是:
在這裏插入圖片描述
按照官網給的教程:

sudo sh cuda_10.0.130_410.48_linux.run

在安裝的時候,記得看清選項,不用再安裝顯卡驅動了

安裝完後:

sudo gedit ~/.bashrc 

在文件最後加上:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64”
export CUDA_HOME=/usr/local/cuda
export PATH="$CUDA_HOME/bin:$PATH"

保存修改,並更新配置文件:

source ~/.bashrc 

打開新終端,檢查Cuda是否安裝成功:

nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130

cuDNN 7.5.0

先從官網上下載,本人選擇的是:
在這裏插入圖片描述

  • 假設下載完後解壓縮至/home/admin/Downloads/CUDA,那麼在/home/admin/Downloads/CUDA/cudnn-10.0-linux-x64-v7.5.0.56/cuda/下有includelib64兩個文件夾
  • Cuda默認安裝位置/usr/local/cuda-10.0
sudo cp -r /home/admin/Downloads/CUDA/cudnn-10.0-linux-x64-v7.5.0.56/cuda/*  /usr/local/cuda-10.0/

檢查cuDNN是否安裝成功
輸入如下命令:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章