首页 > 解决方案 > 如何计算训练时所需的 GPU 内存

问题描述

伙计们!我想计算训练时需要的GPU显存,模型配置如下,右边是我的计算过程。但是当我运行程序时,GPU显存使用了1701M。这与我的计算有很大的不同。 机型配置

使用的 GPU 内存

标签: tensorflowmemorygpunvidiaconv-neural-network

解决方案


推荐阅读