پشتیبانی tensorflow از gpu

Question

سلام. آیا راهی وجود داره برای اینکه متوجه بشیم ایا تنسورفلوی نصب شده ی فعلی بر روی سیستم gpu ساپورت می کنه یا cpu؟

Answer 1

پیدا کردم تنسورفلو تابعی داره برای اینکار به نام is_gpu_available که به صورت زیر می تونید استفاده کنید ر:

import tensorflow as tf

if tf.test.is_gpu_available():
  print("GPU")
else :
  print("CPU")