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 */
此篇文章为自己学习所用笔记,欢迎交流。