غیر فعال سازی GPU در تسورفلو

Question

درحالت آموزش شبکه Alexnet از gpu استفاده کردم درحالت inference باید از cpu استفاده کنم آیا راهی هست که با اون به tensorflow بفهمونیم که session راحتما روی cpu اجرا کنه؟

Answer 1

سلام زمانیکه که session را می سازید config مشخص کنید و توی آن device_count مربوط به GPU را صفر بدید به صورت زیر :

sess = tf.Session(config=tf.ConfigProto(device_count={'GPU': 0}))

 

راه دیگری هم وجود داره که تو کل برنامه gpu را غیر فعال کنید:

import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"    
import tensorflow as tf