[Python] Pytorch GPU 사용 확인

우선 그래픽 드라이버, cuda, cudnn 설치가 완료되었다고 가정한다. 

본인은 아래 블로그를 참고하여 cuda 설치를 하였다. 

[Pytorch] window cuda v9.0, pytorch1.2.0 에서 cuda v10.1, pytorch 1.5.0 설치하기

 

 

 

가상환경을 새로 만들어 프로젝트를 진행함에 있어, 해당 환경에서의 torch 설치가 필요했다. 

이는 pytorch.org에서의 command를 본인의 환경에 맞추어 터미널에서 실행하면 된다. 

Start Locally

 

 

 

실행 코드

이후 pytorch가 제대로 작동되는지 확인하기 위해 아래의 코드를 실행한다.

세 코드 중 하나만 실행하면 되고, 사용가능여부만 확인하려면 첫번째 코드를 실행하면 된다. 

import torch

torch.cuda.is_available() # True
torch.cuda.get_device_name(0) # 'NVIDIA GeForce RTX 3060'
torch.cuda.device_count() # 1