ubuntu系统NVIDIA显卡驱动+cuda10.2+cudnn8.0.5安装

ubuntu18.04系统

设置ubuntu环境
显卡驱动+cuda10.2+cudnn8.0.5
(需要联网)
更新本地软件:

sudo apt-get update

查看系统推荐驱动:

ubuntu-drivers devices


后缀recommended为推荐驱动。
终端输入:

sudo apt-get install nvidia-driver-xxx


即可安装对应驱动。
重启电脑,终端输入:

nvidia-smi


查看安装是否成功。

cuda下载地址,选择对应版本
cuda下载网址
以10.2为例,分别输入:

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


弹出界面选择continue
输入accept
取消驱动选项(前面已经安装完成了,不必重复安装)
等待完成安装。

完成安装之后,要配置环境变量。
打开环境变量文件:

gedit ~/.bashrc


在代码底部添加:

export PATH="/usr/local/cuda-10.2/bin:$PATH" 
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH" 


保存关闭,退出。
在终端更新环境变量:

source ~/.bashrc


终端输入

nvcc -V


验证是否安装成功。

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89


显示这个就是成功了。

安装cudnn
官网:官网
选择需要的版本,选择cuDNN Library for Linux[x86]下载,这时需要登陆,如果你没有账号,注册一个。
解压下载的安装包,进入看到cuda文件夹,在此页面进入终端。
输入以下代码复制文件到对应文件夹:

sudo cp cuda/include/* /usr/local/cuda/include
sudo cp include/* /usr/local/cuda/include/
sudo chmod a+r /usr/local/cuda/lib64/*
sudo chmod a+r /usr/local/cuda/include/*


这里为了防止报错,用代码把整个头文件都复制了。
完成后可输入指令:

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


检测是否安装完成。
安装成功界面:
 

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 5
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#endif /* CUDNN_VERSION_H */

此篇文章为自己学习所用笔记,欢迎交流。

猜你喜欢

转载自blog.csdn.net/weixin_68583698/article/details/131454864