Your browser does not seem to support JavaScript. As a result, your viewing experience will be diminished, and you may not be able to execute some actions.
Please download a browser that supports JavaScript, or enable it if it's disabled (i.e. NoScript).
GPU환경에서 gpu를 사용하는 pytorch을 이용하고 싶은데 하는 방법을 잘 모르겠습니다
안녕하세요.
현재 베타 서비스 중인 GPU 컨테이너에는 GPU를 지원하는 Pytorch가 이미 설치되어 있습니다. 아래 코드로 pytorch가 GPU를 지원하는지 확인할 수 있습니다.
import torch # GPU 지원 여부 print(torch.cuda.is_available()) # GPU 이름 print(torch.cuda.get_device_name(0))