GPU환경에서 gpu를 사용하는 pytorch로 설치하는 방법을 모르겠습니다

[Jaehun Cho]

GPU환경에서 gpu를 사용하는 pytorch을 이용하고 싶은데 하는 방법을 잘 모르겠습니다

[joony.kwon]

안녕하세요.

현재 베타 서비스 중인 GPU 컨테이너에는 GPU를 지원하는 Pytorch가 이미 설치되어 있습니다. 아래 코드로 pytorch가 GPU를 지원하는지 확인할 수 있습니다.

import torch

# GPU 지원 여부
print(torch.cuda.is_available())

# GPU 이름
print(torch.cuda.get_device_name(0))