pipeline加载TensorFlow架构模型,怎么指定TensorFlow使用gpu显存大小?

pipeline加载TensorFlow架构模型,怎么指定TensorFlow使用gpu显存大小?我在12G的卡上运行BSHM人像抠图模型,占用11G,在24G的卡上运行,占用20G,请问怎么设置,其占用显存大小?

展开
收起
游客zxjxornoo67tk 2024-03-05 17:17:49 95 分享 版权
0 条回答
写回答
取消 提交回答

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

热门讨论

热门文章

还有其他疑问?
咨询AI助理