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

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

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

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

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

相关电子书

更多
使用TensorFlow搭建智能开发系统自动生成App UI 立即下载
从零到一:IOS平台TensorFlow入门及应用详解 立即下载
从零到一:IOS平台TensorFlow入门及应用详解(附源 立即下载