如何在AI Studio中查看网络实际显存占用量?
收藏
直接在AI Studio上使用"性能监控"模块是没办法查看到网络的实际显存占用量的。
今天查文档看到PaddlePaddle在分配requested_size大小的显存时,先定义一个显存池的大小,记为chunk_size,chunk_size由环境变量 FLAGS_fraction_of_gpu_memory_to_use确定,表征chunk_size在全部显存的占比,默认值为0.92,即框架预先分配显卡92%的显存。
因此在AI Studio上使用“性能监控”模块没办法看到网络的实际显存占用量。
0
收藏
请登录后评论
在想测量网络的实际显存占用量时,可以设置该占比为0,观察nvidia-smi显示的显存占用情况。或使用"性能监控"模块来查看显存占用。
经典版的可视化显存一直bug哈哈哈
只能终端用nvidia-smi看