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() ..
머신러닝 딥러닝/Tools
2022. 1. 18. 16:38
