pytorch 설치하기

 

 

flair모델을 튜닝하면서 gpu를 지원하는 pytorch를 설치하고 과정을 기록해둔다.

 

 

https://pytorch.kr/get-started/locally/

파이토치 한국 사용자 모임 (PyTorch Korea User Group)

 

 

pytorch설치

 

 

위 링크에서 원하는 빌드 환경을 선택하면 pytorch 설치 명령어가 나온다. 

데스크탑에 cuda 12.6 버전이 설치되어 있어서 12.4로 선택하였다. 

 

설치이후 제대로 설치되었나 확인을 해본다. 

import torch


print(torch.__version__)
print(torch.version.cuda)  # 설치된 CUDA 버전 확인
print(torch.backends.cudnn.enabled)  # cuDNN이 활성화되었는지 확인


# GPU 사용 가능한지 확인
if torch.cuda.is_available():
    print("CUDA is available. Using GPU.")
else:
    print("CUDA is not available. Using CPU.")

 

 

2.4.1+cu124
12.4
True
CUDA is available. Using GPU.

 

 

 

 

제대로 설치가 되었고 이제 모델 튜닝시 gpu사용이 가능하다. 

flair는 자동으로 gpu 인식하고 사용해서 코드에서 따로 변경을 할 필요는 없다.

 

기존 코드 돌려보니까 체감상 10배정도는 빨라진듯 ㅠㅠ

 

gpu 만세..!!!!