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
참고로, 구버전으로 다운그레이드하는 경우도 진행이 가능합니다.
'Python > Pytorch' 카테고리의 다른 글
[Pytorch] 원소 반복 텐서 확장 : expand, expand_as, repeat 사용법 (0) | 2022.01.26 |
---|---|
[Pytorch] squeeze와 unsqueeze 함수 사용법 정리 (2) | 2022.01.25 |
[Pytorch] 파이토치 텐서 변환, 선언하는 방법 (0) | 2021.11.14 |