CPU/GPU常用查看命令操作

GPU相关操作

【1】查看GPU使用情况语句（GPU）

watch -n 1 nvidia-smi

【2】指定显卡

import os os.environ["CUDA_VISIBLE_DEVICES"] = "0"

【3】限制GPU用量

1. 设置使用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