پیدا کردم تنسورفلو تابعی داره برای اینکار به نام is_gpu_available که به صورت زیر می تونید استفاده کنید ر:
import tensorflow as tf if tf.test.is_gpu_available(): print("GPU") else : print("CPU")