文章目录

• 1. 安装nvidia驱动；
• 2. 安装cuda;
• 3.安装cudnn

首先，保证有一个显卡，并且工作正常，狗头。

1. 安装nvidia驱动；

执行下面命令，系统会告诉你，需要什么驱动;

ubuntu-drivers devices
执行下面命令，安装系统推荐驱动；
sudo ubuntu-drivers autoinstall

安装完成后，执行命令：

nvidia-smi

查看显卡的工作情况；

2. 安装cuda;

选择适合的cuda版本，这点往往是由你要执行的任务决定的。

下载连接：
https://developer.nvidia.com/cuda-downloads
选择你的型号，按照说明执行安装即可，建议选择runfile。
如图，执行对应命令，安装完成。

这里，安装cuda10.02，ubantu版本，runfile(local)。
执行如下命令：

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

如果报gcc版本冲突，可能需要gcc降级，参考连接：ubantu 20.04 gcc和cuda版本冲突

具体细节，可参考Ubuntu 18.04安装CUDA和cuDNN。

3.安装cudnn

下载链接，cudnn下载链接，这个需要登录验证，选择对应的版本即可，。可参考链接：ubantu安装CUDA和cudnn，但这个可能在查看cudnn版本时有一个坑。

这里贴出安装的命令：
在下载并解压后，在解压后的文件夹内终端执行：

sudo cp cuda/include/* /usr/local/cuda-10.2/include/

sudo cp cuda/lib64/* /usr/local/cuda-10.2/lib64/

sudo chmod +x /usr/local/cuda-10.2/include/cudnn.h

sudo chmod +x /usr/local/cuda-10.2/lib64/libcudnn*

完成后，查看版本：
可以尝试

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

没有显示，可以执行如下命令：

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

操作截图：

安装完成。