What's the real meaning of FLAGS_fraction_of_gpu_memory_to_use?
Created by: imistyrain
P40单卡显存24445MiB,paddle-gpu 1.2,且没有其他任务占用显存. 使用models模型里的image calssification训练alexnet 运行的命令为:
python train.py \
--model=AlexNet \
--batch_size=64 \
--total_images=1281167 \
--class_dim=1000 \
--image_shape=3,224,224 \
--model_save_dir=output/ \
--with_mem_opt=False \
--lr_strategy=piecewise_decay \
--num_epochs=120 \
--lr=0.01 \
当设置FLAGS_fraction_of_gpu_memory_to_use=0.01时,占用显存1773MiB 当设置FLAGS_fraction_of_gpu_memory_to_use=0.1时,占用显存2693MiB 当设置FLAGS_fraction_of_gpu_memory_to_use=0.2时,占用显存5015MiB 更特别的 当设置为FLAGS_fraction_of_gpu_memory_to_use=0时,占用显存在1019MB到1635MB之间波动 有点迷惑了,这个标志的真正含义到底是啥呢?从字面意思是预先申请占用的百分比,但实际结果好像并不是.