tesorflow GPU 확인 방법

 

tensorflow에서 GPU를 제대로 로딩하는지 확인 하는 몇가지 방법이다. 어느 것으로 하든 GPU가 제대로 보인다면 설치가 정상적으로 된 것이고 GPU를 이용해서 모델을 학습할 수 있다는 것을 의미한다.

 

1. list_local_devices() 

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))

 

 

 

2. list_physical_devices('GPU')

tf.config.list_physical_devices('GPU')

 

 

3. is_build_with_cuda()

import tensorflow as tf
print(tf.test.is_built_with_cuda())

output : True

 

 

4. gpu_device_name()

tf.test.gpu_device_name()

output : '/device:GPU:0'

 

 

연관글

2022.01.09 - [Linux] - [ubuntu 서버만들기 #2] nvidia graphic driver 설치

2022.01.09 - [Linux] - [ubuntu 서버만들기 #3] cuda 11.2 & cud 8.1 설치하기

2022.01.18 - [머신러닝 딥러닝/Tools] - conda 환경에서 tensorflow에서 GPU 인식 못하는 문제