Linux 中使用 Docker 安装 PaddleOCR

发表于 2023-12-27 分类于 System ， Linux 本文字数： 1.9k 阅读时长 ≈ 2 分钟

本文将介绍如何在 Ubuntu 的 Docker 中安装 PaddleOCR，并启用 Cuda 进行加速。

使用 Docker 安装 PaddlePaddle 环境

# 进入到用户目录，创建一个目录新目录
cd ~ && mkdir dockerMount/paddleOCR && cd ./dockerMount/paddleOCR

# 创建一个容器，使用CUDA12+CUDNN8.9
# 设置docker容器共享内存shm-size为32G，建议设置32G以上
sudo docker run --gpus all --name ppocr -v $PWD:/paddle --shm-size=32G --network=host -it registry.baidubce.com/paddlepaddle/paddle:2.6.0-gpu-cuda12.0-cudnn8.9-trt8.6 /bin/bash

可使用 exit 退出容器。

可以通过下列命令重新进入容器：

# 完整方式
sudo docker container exec -it ppocr /bin/bash
# 简化方式
sudo docker exec -it ppocr bash

至此，PaddlePaddle 环境就安装成功了

特别说明：

官方教程使用的是 nvidia-docker 来安装的，NVIDIA/nvidia-docker: 目前被 NVIDIA/nvidia-container-toolkit 代替，因此，在使用时，将 nvidia-docker 换成 docker --gpu all 即可。

测试 paddlepaddle 安装结果

进入 python shell，然后执行 paddle.utils.run_check() 查看状态

1
2
3

python
import paddle
paddle.utils.run_check()

正常的输出结果如下：

Running verify PaddlePaddle program ...
I0108 07:52:51.569307   893 program_interpreter.cc:212] New Executor is Running.
W0108 07:52:51.569667   893 gpu_resources.cc:119] Please NOTE: device: 0, GPU Compute Capability: 8.9, Driver API Version: 12.2, Runtime API Version: 12.0
W0108 07:52:51.570616   893 gpu_resources.cc:164] device: 0, cuDNN Version: 8.8.
I0108 07:52:51.671880   893 interpreter_util.cc:624] Standalone Executor is Used.
PaddlePaddle works well on 1 GPU.
PaddlePaddle is installed successfully! Let's start deep learning with PaddlePaddle now.