easyOCR 사용을 위한 CUDA 셋팅

2023.01.15 - [전자기기 리뷰] - 레노버 p14s Gen3 노트북 CUDA 설치하기

이전 글에서 CUDA를 설치하는 것까지 했다.

이후 easyOCR사용을 위한 추가 셋팅을 기록한다.

 

1. easyOCR 설치

python -m pip install easyocr

터미널에서 easyOCR을 설치한다.

 

2. CUDA 툴킷 설치

https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html

혹시 놓친게 있나 해서 CUDA 사용을 위한 튜토리얼 문서를 봤다.

문서 중간쯤에 CUDA 툴킷 기본요소 설치를 해야한다고 나와서 설치했다.

 

이런식으로 CUDA 관련된 여러가지 패키지가 설치된다.

conda install cuda -c nvidia

이건 관리자 권한으로 실행해야해서 conda prompt를 관리자 권한으로 실행해서 설치했다.

 

3. pytorch 설치

https://pytorch.org/get-started/previous-versions/

easyOCR은 pytorch framework로 만들어졌다.

그래서 CUDA 사용이 가능한 버전의 pytorch를 설치해야 한다.

 

opencv 버전 문제가 있다고 하는데... 음....

나는 CUDA 11.2 버전이기 때문에 하나 낮은 11.1 기준으로 설치했다.

여러개 시도해봤는데, 오류없이 설치되는게 torch 1.9.1 버전이었다.

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html 

 

4. 성공

드디어 성공이다.

코드상에 CPU 대신 GPU 쓰라고 뜨는것도 없고,

T550 그래픽카드가 열심히 일하는 모습을 볼 수 있다.

 

실제 작업 속도도 눈에 띄게 빨라졌다.

 

이상으로 CUDA 셋팅을 마무리한다.