본문 바로가기

Language

[Python] Tensorflow로 사용가능한 GPU 확인하기

728x90

Tensorflow를 수행할 때 GPU는 꼭 필요합니다.

Python에서 현재 사용가능한 GPU가 무엇인지 확인하려면 다음의 명령을 활용하시면 됩니다.

from tensorflow.python.client import device_lib

device_lib.list_local_devices()

위 명령을 수행하시면 현재 사용가능한 GPU에 대한 정보가 출력됩니다.

 

반응형