如何在AI Studio中查看网络实际显存占用量?
收藏
快速回复
AI Studio平台使用 其他其他 2039 2
如何在AI Studio中查看网络实际显存占用量?
收藏
快速回复
AI Studio平台使用 其他其他 2039 2
直接在AI Studio上使用"性能监控"模块是没办法查看到网络的实际显存占用量的。
今天查文档看到PaddlePaddle在分配requested_size大小的显存时,先定义一个显存池的大小,记为chunk_size,chunk_size由环境变量 FLAGS_fraction_of_gpu_memory_to_use确定,表征chunk_size在全部显存的占比,默认值为0.92,即框架预先分配显卡92%的显存。
因此在AI Studio上使用“性能监控”模块没办法看到网络的实际显存占用量。
0
收藏
回复
全部评论(2)
时间顺序
DeepGeGe
#2 回复于2021-12

在想测量网络的实际显存占用量时,可以设置该占比为0,观察nvidia-smi显示的显存占用情况。或使用"性能监控"模块来查看显存占用。

0
回复
Y_kira
#3 回复于2021-12

经典版的可视化显存一直bug哈哈哈

只能终端用nvidia-smi看

0
回复
在@后输入用户全名并按空格结束,可艾特全站任一用户