GPU佔顯卡很大內存時,設置限制GPU的用量

當顯卡內存飽滿時   指定顯卡session 一、設置使用GPU的百分比 import tensorflow as tf import keras.backend.tensorflow_backend as KTF #進行配置,使用30%的GPU config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fractio
相關文章
相關標籤/搜索