NVIDIA-SMI命令系列详解

nvidia-smi:显示所有GPU的当前信息状态

Memory-Usage: 显存使用率,100MiB/8105miB,即显存总量为8105miB,当前显存总占用为100MiB。
Volatile GPU-Util: GPU使用率

Processes:
PID:所提任务id
Process name:任务名
GPU Memory Usage:该任务所用的显存量

可在脚本里通过以下命令设置任务占的显存:

config.gpu_options.allow_growth = True
#分配给tensorflow的GPU显存大小为:GPU实际显存*0.2
config.gpu_options.per_process_gpu_memory_fraction = 0.2 

猜你喜欢

转载自blog.csdn.net/leitouguan8655/article/details/86672405