تخصيص ال GPU لتنفيذ مهمة معينة في كيراس Keras وتنسرفلو Tensorflow

يمكنك تعيين متغيرات environment في ال notebook باستخدام os.environ. حيث يجب أن تقوم بما يلي قبل تهيئة TensorFlow وذلك لقصر TensorFlow على GPU الأول (جعله يستخدم أول Gpu فقط):

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0"
#وتريد تفعيل عدد محدد منها Gpu في حالة كان لديك 4 وحدات 
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0,2,3" #هنا Gpu نضع مانريده من ال 

وبالتالي سيتم قصر تنسرفلو على أول gpu فقط، حيث سيتم إخفاء الثاني عنه، وللتأكد يمكنك استخدام الكود التالي:

import notebook_util
notebook_util.pick_gpu_lowest_memory()
import tensorflow as tf
# أو
from tensorflow.python.client import device_lib
print device_lib.list_local_devices()

حيث سيعرضان لك قائمة الأجهزة المرئية (غير المخفية).