悟能的师兄_个人页
悟能的师兄
文章
0
问答
1
视频
0
个人介绍
暂无个人介绍
擅长的技术
- Linux
暂无更多信息
2024年12月
-
12.03 10:36:34
回答了问题
2024-12-03 10:36:34
ModelScope模型如何指定gpu运行呢？
赞0 踩0 评论0
暂无更多信息
-
回答了问题
2024-12-03
ModelScope模型如何指定gpu运行呢？
真的服了，这from modelscope import pipeline,不管用Bert还是lert都很简单指定到cuda:0,cuda:1就可以了，这玩意儿设定了device，传参数进去却不管用，参考参考国际惯例OK，自己拍脑袋到飞起的设定，别人怎么用你家的产品啊。只能加代码转移，显存却释放不了，算了算了，不跟你们家玩了将模型显式地移动到 GPU1model = pipeline.modelmodel.to(device_obj)赞0 踩0 评论0
滑动查看更多
暂无更多信息