Docker 사용방법(CUDA, paddleOCR 설치)

이 글은 우분투 22.04를 기준으로 Docker 를 어떻게 사용하는지 적어두는 글이다.

본인은 현재 우분투 22.04에 CUDA 12.4 를 사용하고 있으며, 현재 paddle OCR 을 사용하려면 CUDA 11.8 이 필요하므로 이를 docker 를 사용하여 설치해보려 한다.(처음 해봄)

먼저, CUDA 11.8 버전이 필요하다.(현재 12.4버전) 따라서 docker 로 먼저 CUDA환경을 설치해주자.

1. NVIDIA Container Toolkit 설치

먼저, NVIDIA Container Toolkit 을 설치해야 한다. 이건 Docker에서 CUDA GPU를 쓸 수 있게 해주는 다리 역할을 한다.
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
여기 사이트의 설명을 따라가면서 nvidia container toolkit 을 설치하도록 하자.

2. docker 이미지 다운로드

paddle OCR 홈페이지의 빠른 설치 탭에서 맞는 버전을 찾아 코드를 구성하고 터미널에 붙여넣기하자.

이렇게 구성한 docker container의 이름은 코드에 적힌 대로 'paddle'이 된다. 따라서 추후 container 실행 시 다음 코드를 입력하면 된다.

docker start -ai paddle

[container를 실행하고 cuda 버전을 확인해본 모습]

참고-1 container 삭제, 목록 확인 등

• container 삭제

docker rm paddle

• container 목록 확인

docker ps -a

• container 종료

exit

3. paddle OCR 설치

docker 가상 환경을 실행하고, 다음 코드를 이용하여 paddlepaddle 과 paddleOCR 를 설치한다.

pip install paddlepaddle-gpu

pip install paddleocr

이후 실행 가이드는 다음 링크를 참고한다.
https://paddlepaddle.github.io/PaddleOCR/latest/ppocr/quick_start.html#11-paddlepaddle