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


  • Level 2

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

  • Level 6

    안녕하세요.

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

    import torch
    
    # GPU 지원 여부
    print(torch.cuda.is_available())
    
    # GPU 이름
    print(torch.cuda.get_device_name(0))
    

ONLINE USERS

POPULAR TOPICS

SUGGESTED TOPICS

  • 2
  • 1
  • 4
  • 7
  • 2
  • 2
  • 2