pytorch , cuda 버전 확인

인공지능 학습을 위해 Pytorch 와 cuda 를 설치 하고 둘의 호환성 을 위해 버전을 확인해야 합니다.

 



1) Pytorch 버전 확인

 

pytorch 버전 확인을 위해 cmd 에서 가상환경 activate 한 후에 아래와 같이 입력 하면 됩니다.

python

import torch

print(torch.__version__)

 

 

 

2) cuda 버전 확인

 

• cuda 설치 확인
  cuda 설치했을경우 설치가 되었는지 확인 하는 작업이 필요합니다. cmd 창을 열고 입력하세요. ( 가상환경 X )

nvcc -V

• cuda 사용가능 확인
  cuda 와 Pytorch 의 호환성이 매우 중요합니다. cuda 의 사용가능여부를 확인하고, 안될경우 Pytorch( 링크 ) 에 들어가서 버전에 맞는 cuda 를 설치해야 합니다.

python

import torch

torch.cuda.is_available()

위와 같이 True 가 나오면 cuda 사용가능한 상태입니다.

만약 False 가 나오면 cuda 와 Pytorch 의 호환버전을 체크 하셔야합니다.

 

---