GPU 사용 가능 여부 확인하기

작성자 김아름 수정일 2023-02-27 13:42

#SparklingSoDA3.0, #스파클링소다3.0, #텐서플로우, #tensorflow, #gpu

아티클 관련 제품: SparklingSoDA3.0

들어가며

  • tensorflow를 통해 GPU 사용 가능 여부를 알아봅니다.



GPU 사용 가능 여부 확인하기


  • 엔진을 실행시킬 resource를 설정합니다.

이때 GPU를 선택합니다.


* resource 이름은 상이할 수 있습니다.


  • 올라온 엔진에 로그인 합니다.


  • ipynb 파일을 생성합니다.


  • tensorflow 설치 여부를 확인합니다.
!pip list | grep tensorflow


  • GPU 사용 가능 여부를 확인합니다.

아래 코드를 실행 후, GPU 가 있다면 gpu 사용이 가능합니다.

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())


만일 gpu 사용이 불가할 때는 아래와 같이 CPU만 출력됩니다.



마무리

  • tensorflow를 사용하여 GPU 사용 가능 여부를 확인해 보았습니다.

아티클이 유용했나요?

훌륭합니다!

피드백을 제공해 주셔서 감사합니다.

도움이 되지 못해 죄송합니다!

피드백을 제공해 주셔서 감사합니다.

아티클을 개선할 수 있는 방법을 알려주세요!

최소 하나의 이유를 선택하세요
CAPTCHA 확인이 필요합니다.

피드백 전송

소중한 의견을 수렴하여 아티클을 개선하도록 노력하겠습니다.

02-558-8300