Ubuntu20.04安装cuda10.2和cudnn7.6.5

结果展示:
Ubuntu20.04安装cuda10.2和cudnn7.6.5_第1张图片
Ubuntu20.04安装cuda10.2和cudnn7.6.5_第2张图片

禁用nouveau

首先确保禁用显卡自带驱动。

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

在文件中添加

blacklist nouveau
options nouveau modeset=0

wq关闭保存,然后执行

sudo update-initramfs -u

重启

sudo reboot

重启,执行下面语句,如没有反馈信息则禁用完成。

lsmod | grep nouveau

查看显卡型号

lshw -numeric -C display

安装驱动,去官网下载想要安装的版本https://www.geforce.cn/drivers ,要确保之前显卡未安装驱动,安装

sudo chmod a+x NVIDIA-Linux-x86_64-470.82.run
sudo ./NVIDIA-Linux-x86_64-470.82.run --no-opengl-files --no-x-check --no-nouveau-check

–no-opengl-files 表示只安装驱动文件,不安装OpenGL文件。
–no-x-check 安装驱动时不检查X服务。
–no-nouveau-check 安装驱动时不检查nouveau。
检查驱动安装

nvidia-smi

安装CUDA

去官网下载想要的cuda版本,https://developer.nvidia.com/cuda-toolkit-archive ,本次选择10.2。CUDA10.2要求gcc版本为7,ubuntu自带9,需要安装gcc7。

sudo apt install gcc-7 g++-7

查看gcc版本

ls /usr/bin/gcc*

请添加图片描述

对gcc进行默认切换

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 50
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100

数值越高,优先权越大。
gcc默认版本

sudo update-alternatives --config gcc

Ubuntu20.04安装cuda10.2和cudnn7.6.5_第3张图片

输入数字1,切换完成。
下载cuda10.2

wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

安装依赖工具

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

安装cuda

sudo sh ./cuda_10.2.89_440.33.01_linux.run

按下键,选择continue。
输入accept。
DRIVRER为驱动,按空格键,取消安装!!!
按下键,选择install,回车。
修改环境变量

gedit ~/.bashrc

添加

export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source

source ~/.bashrc

cuda安装完成。
检查cuda信息语句

nvcc -V

Ubuntu20.04安装cuda10.2和cudnn7.6.5_第4张图片

安装cudnn

从 https://developer.nvidia.com/cudnn 下载cuda版本对应的cudnn.
解压

tar -xvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

复制文件

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

安装完成。
查看命令:

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

Ubuntu20.04安装cuda10.2和cudnn7.6.5_第5张图片
完成!

你可能感兴趣的:(linux,python)