开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

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

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

展开
收起
游客zxjxornoo67tk 2024-03-05 17:17:49 53 0
0 条回答
写回答
取消 提交回答

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

相关电子书

更多
DeepStream: GPU加速海量视频数据智能处理 立即下载
阿里巴巴高性能GPU架构与应用 立即下载
GPU在超大规模深度学习中的发展和应用 立即下载