如何在AI Studio中查看网络实际显存占用量？

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

项目

数据集

课程

比赛

模型库

活动

论坛

访问飞桨官网

DeepGeGe 发布于2021-12

直接在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

回复

时间顺序
时间逆序