Language
[Python] Tensorflow로 사용가능한 GPU 확인하기
까레까레
2021. 4. 6. 01:43
728x90
Tensorflow를 수행할 때 GPU는 꼭 필요합니다.
Python에서 현재 사용가능한 GPU가 무엇인지 확인하려면 다음의 명령을 활용하시면 됩니다.
from tensorflow.python.client import device_lib
device_lib.list_local_devices()
위 명령을 수행하시면 현재 사용가능한 GPU에 대한 정보가 출력됩니다.
반응형