2021 年最新版 Ubuntu 20.04 配置 Caffe 深度学习环境

最新推荐文章于 2024-05-11 23:32:26 发布

原创

最新推荐文章于 2024-05-11 23:32:26 发布 · 5.3k 阅读

37 ·

CC 4.0 BY-SA版权

文章标签：

#linux #ubuntu #caffe #深度学习

本文详细介绍了如何在 Ubuntu 20.04 上配置 Caffe 深度学习环境，包括更新系统软件包、安装 NVIDIA 显卡驱动和 CUDA 工具包、安装 OpenCV 3，以及解决 Caffe 与 OpenCV 4 兼容性问题。最后，文章还展示了如何安装 Caffe 框架并进行 MNIST 数据集的训练。

前言

本文写于 2021 年 7 月 15 日。若按本文的方法配置出现了错误，请留意是否距离写作时间过远。

请确保你的电脑显卡是 NVIDIA 显卡。本文不适用于 AMD 显卡和 Intel 集成显卡。

一、更新系统软件包

在终端中执行如下命令，将系统内核和基础组件升级至最新版本，否则后续编译会因为组件版本问题遇到不可预料的错误。

# 更新软件仓库缓存
sudo apt-get update

# 更新所有软件包至最新版本
sudo apt-get upgrade -y

# 更新的过程中可能会升级 Linux 内核，执行下述命令删除不再使用的旧内核
sudo apt-get autoremove --purge -y

# 重启
reboot

二、安装 NVIDIA 显卡驱动

这里使用 ubuntu-drivers 命令安装软件仓库里的 NVIDIA 驱动。这种方法不但简单，而且安装上的驱动也很稳定，不会导致系统崩溃。

# 自动安装推荐的驱动
sudo ubuntu-drivers autoinstall

# 重启
reboot

重启后，输入 nvidia-smi 命令，检查 NVIDIA 驱动和 CUDA 驱动是否被正确安装。如果有类似下面这样的输出，则说明安装成功。

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.84       Driver Version: 460.84       CUDA Version: 11.2     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  GeForce GT 720      Off  | 00000000:01:00.0 N/A |                  N/A |
| 40%   41C    P0    N/A /  N/A |    146MiB /   974MiB |     N/A      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+