Python/Pytorch

[Pytorch] 쿠다 버전 확인, 파이토치 버전 체크, 업데이트 방법

jimmy_AI 2022. 1. 23. 16:01
반응형

CUDA 및 파이토치 버전 확인 및 관리

Pytorch를 사용하는 경우 CUDA 버전을 확인하고,

쿠다와 호환이 잘 되는 파이토치 버전으로 변경해주어야 하는

경우가 있을 수 있습니다.

 

이번 글에서는 간략하게 파이토치를 최적으로 사용하기 위한

환경에 대해 점검하고 버전을 관리하는 방법을 말씀드려보도록 하겠습니다.

 

 

쿠다 버전 확인법

CUDA 버전을 확인하는 방법은 명령 프롬프트에 $ nvcc --version를 입력하시거나,

ipynb 파일의 셀 내에 !nvcc --version를 입력해주시면 됩니다.

출력 결과에서 마지막 줄의 숫자를 확인해보시면 됩니다.

 

예시로 코랩 환경에서 CUDA의 버전을 조회해본 결과,

11.1 version을 사용하고 있는 것을 알 수 있었습니다.

 

혹은, 아래의 pytorch version을 체크하는 방법으로도 같이 조회가 가능합니다.

 

 

파이토치 버전 확인법

다른 모듈의 version 체크 방법처럼

torch 모듈을 import 해주신 뒤, __version__ 속성으로 조회해주시면 됩니다.

import torch

torch.__version__
# 1.10.0+cu111

pytorch는 1.10.0을 사용 중이었으며,

cu111인 것을 보아 11.1 버전의 CUDA를 사용하고 있다는 점도 확인이 가능했습니다.

 

해당 CUDA와 호환이 잘되는 torch의 버전 목록은 아래 사이트에서 확인이 가능합니다.

https://download.pytorch.org/whl/cu111/torch_stable.html

 

다른 쿠다의 경우, url의 가운데 cu111 부분을

9.2에서는 cu92, 10.1에서는 cu101처럼 바꿔주시면 됩니다.

 

 

파이토치 업그레이드 방법

원하는 버전으로 torch 모듈을 업데이트하는 방법은

아래의 명령어를 명령 프롬프트나 ipynb의 셀 내에 입력해주시면 됩니다.

pip install torch --upgrade version==1.10.1

참고로, 구버전으로 다운그레이드하는 경우도 진행이 가능합니다.