安装显卡驱动
cuda和graphic driver版本对应关系
一般根据需要,安装指定的cuda版本,因此显卡驱动的版本需要满足最低的版本要求。

查看可安装的驱动版本
ubuntu-drivers devices
自动安装推荐的驱动版本
如果推荐版本满足要求,可以自动安装推荐版本的驱动。
sudo ubuntu-drivers autoinstall
安装后重启
检查驱动是否安装成功
nvidia-smi
显示最高可安装的cuda版本为10.2,当前驱动版本为440。

安装CUDA10.0
下载cuda
参考cuda安装方法,进入cuda官方下载链接下载安装即可;
安装cuda
由于已经有了显卡驱动,不需要进行安装; cuda安装在默认路径即可;
sudo sh cuda_10.0.130_410.48_linux.run

查看安装的cuda版本
cat /usr/local/cuda-10.0/version.txt

检测是否安装成功
nvcc -V

安装cudnn
cudnn是GPU加速计算深层神经网络的库。
下载cudnn
参考cudnn安装方法,进入cudnn官方下载链接; 选择cuda10对应的cudnn版本,点击cudnn library for linux进行下载,注意需要注册登陆才能下载;

安装cudnn
解压
tar -xvf cudnn-10.0-linux-x64-v7.6.5.32.tgz
依次执行命令
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64
sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h /usr/local/cuda-10.0/lib64/libcudnn*
查询cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
 cudnn安装成功。
[1]:Ubuntu18.04 安装Cuda 10.0 和Cudnn [2]:Ubuntu18.04安装cuda(10.1及10.0)和cudnn [3]:cuda toolkit documentation [4]:cuda官方下载链接 [5]:cudnn官方下载链接
|