GPU相关操作
【1】查看GPU使用情况语句(GPU)
watch -n 1 nvidia-smi
【2】指定显卡
import os os.environ["CUDA_VISIBLE_DEVICES"] = "0"
【3】限制GPU用量
- 设置使用GPU的百分比
import tensorflow as tf
import keras.backend.tensorflow_backend as KTF #进行配置,使用30%的GPU
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3
session = tf.Session(config=config) # 设置session KTF.set_ses