tensorflow gpu 사용여부 확인

han811·2020년 12월 12일
0

tensorflow 2.x

목록 보기
1/2
post-thumbnail

다음의 코드를 실행하면 사용 가능한 장치 목록이 나오고 이중 GPU로 되어 있는 부분이 있으면 사용가능한 상태이다.

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

profile
han811

0개의 댓글