pipeline加载TensorFlow架构模型，怎么指定TensorFlow使用gpu显存大小？
pipeline加载TensorFlow架构模型，怎么指定TensorFlow使用gpu显存大小？我在12G的卡上运行BSHM人像抠图模型，占用11G，在24G的卡上运行，占用20G，请问怎么设置，其占用显存大小？
展开
收起
相关问答
热门讨论
热门文章
ModelScope中，模型下载默认路径在哪个路径？
3925
com/action/joingroup?code=v1是什么意思
4634
服务器上onnxruntime-gpu 调用结束，如何释放显存
3721
我希望通过damo-YOLO训练1500*1500的图片
7531
请问在 ModelScope 上的模型断网使用报这个错误啥原因了?
1144
streamlit 生成链接无法访问，gradio只生成内网，无法生成公网
1981
Modelscope把模型指定下载到 e盘的某个文件夹 需要如何设置呀？
1270
在ModelScope中，请问模型下载到本地的具体方法？
898
modelscope cache 环境变量是啥？
2027
请问一下ModelScope这个报错该怎么解决？
724
展开全部
AI人像动漫
864
Stable Diffusion 介绍与入门
1358
关键点检测从入门到进阶
865
《YOLOv8原创自研》专栏介绍 & CSDN独家改进创新实战&专栏目录
56
一种基于YOLOv8改进的高精度红外小目标检测算法 （原创自研）
49
YOLOv10实战：红外小目标实战 | 多头检测器提升小目标检测精度
53
图像分类基础与实战
510
modelscope调用的模型如何指定在特定gpu上运行？排除使用CUDA_VISIBLE_DEVICES环境变量
716
YOLOv10实战：SPPF原创自研 | SPPF_attention，重新设计加入注意力机制 | NEU-DET为案列进行展开
45
人像抠图
895
展开全部
