1. Python 환경 설정
먼저 Python 3.8.5를 다운로드하여 설치합니다:
wget https://mirrors.huaweicloud.com/python/3.8.5/Python-3.8.5.tgz
다운로드 완료 후 압축을 풀고, 컴파일 및 설치를 진행합니다.
2. PaddlePaddle GPU 버전 설치
참고 문서:
- 간편 가이드: PaddleOCR Quick Start
- 공식 문서: PaddlePaddle 설치
GPU 버전을 사용하려면 CUDA가 필요합니다. 먼저 CUDA 버전을 확인합니다:
[root@tx-sh-orc-001 ocr]# nvidia-smi
Fri Jun 10 14:46:23 2022
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 410.104 Driver Version: 410.104 CUDA Version: 10.0 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 Tesla P40 On | 00000000:00:08.0 Off | 0 |
| N/A 34C P0 50W / 250W | 12015MiB / 22919MiB | 0% Default |
+-----------------------------------------------------------------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 22590 C /usr/local/Python-3.8.5/bin/python3 12005MiB |
+-----------------------------------------------------------------------------+
현재 CUDA Version: 10.0으로 상당히 구버전입니다. 최신 PaddlePaddle은 최소 CUDA 10.1을 요구하므로 두 가지 해결책이 있습니다:
- CUDA를 10.1~11.2 사이로 업그레이드 (NVIDIA 공식 아카이브: CUDA Toolkit Archive)
- 구버전 PaddlePaddle 설치
CUDA 10.0에 맞는 구버전을 찾아 설치합니다. 브라우저에서 Ctrl + F로 10.0을 검색하여 다음 명령어를 찾습니다:
python -m pip install paddlepaddle-gpu==2.0.2.post100 -f https://paddlepaddle.org.cn/whl/mkl/stable.html
위 명령이 실패할 경우, whl 파일을 직접 다운로드해야 합니다. -f 옵션의 URL(whl 목록)에서 2.0.2.post100을 검색하고, 적합한 파일의 링크를 복사한 후 다음 명령을 실행합니다:
pip3 install https://paddle-wheel.bj.bcebos.com/2.0.2/avx/paddlepaddle_gpu-2.0.2.post100-cp38-cp38-linux_x86_64.whl
이 과정으로 설치의 90%가 완료됩니다.
3. PaddleOCR 설치
CUDA 버전이 낮아 PaddlePaddle 2.0.2를 사용하므로, 호환되는 PaddleOCR 2.0.2를 설치합니다:
pip install "paddleocr==2.0.2"
참고로 이 문서에서 사용된 주요 패키지 버전은 다음과 같습니다:
- Python 3.8.5
- CUDA 10.0
- PaddlePaddle-GPU 2.0.2
- PaddleOCR 2.0.2