上验证cudnn是否安装成功_ubuntu18.04 安装cuda、cudnn、tensorflow和pytorch其实很简单...

目录

一. 安装NVIDIA驱动

1. 首先，查看NVIDIA驱动是否已安装

2.安装NVIDIA驱动

二.安装cuda

1. 查看NVIDIA版本对应的cuda版本

2. 选择CUDA版本（上限原则）

3. 安装cuda

4. 安装cndnn

三.安装tensorflow、pytorch深度学习框架

1. 安装tensorflow-gpu

2. 安装pytorch

四.参考

一. 安装NVIDIA驱动

1. 首先，查看NVIDIA驱动是否已安装

在终端输入以下命令，如果显示以下结果，说明NVIDIA驱动已经正常安装，否则，需要重新安装；

nvidia-smi

2.安装NVIDIA驱动

方法一：使用Ubuntu自带的镜像源安装

第一步：先在软件中搜索“软件和更新”，然后选择一个可用的镜像源，我选择的是清华的镜像源。我的选择如下：

第二步：选择源之后，还是在软件更新中，选择附加驱动，然后选择合适的驱动，应用更改就好,我本机选择的是nvidia-driver-455的驱动。

第三步：完成安装后，在终端输入 nvidia-smi，如果可以显示以下结果，恭喜你，安装NVIDIA驱动成功啦。

方法二：下载NVIDIA官方驱动

第一步： 进入NVIDIA官方下载与之匹配的驱动: https://www.nvidia.cn/geforce/drivers/ ,下图是我本机驱动匹配的信息。点击“开始搜索”，进入NVIDIA驱动下载界面。