안녕하세요. 현재 베타 서비스 중인 GPU 컨테이너에는 GPU를 지원하는 Pytorch가 이미 설치되어 있습니다. 아래 코드로 pytorch가 GPU를 지원하는지 확인할 수 있습니다. import torch # GPU 지원 여부 print(torch.cuda.is_available()) # GPU 이름 print(torch.cuda.get_device_name(0))