EDU LEVEL DEVTH IDE FORUM
  3. GPU환경에서 gpu를 사용하는 pytorch로 설치하는 방법을 모르겠습니다

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


  • 0
    Level 3

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


  • 0
    Level 6

    안녕하세요.

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

    import torch

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

# GPU 이름
print(torch.cuda.get_device_name(0))
Log in to reply
 

ONLINE USERS

POPULAR TOPICS

SUGGESTED TOPICS


  • 구름IDE 관련 •  

    1


  • 구름IDE 관련 •  

    4


  • 구름IDE 관련 •  

    2


  • 구름IDE 관련 •  

    7


  • 구름IDE 관련 •  

    1


  • 구름IDE 관련 •  

    3


  • 구름IDE 관련 •  

    3


  • 구름IDE 관련 •  

    2