ModelScope生产环境 cuda 报错怎么处理？
ModelScope生产环境 cuda 报错怎么处理？ CUDA out of memory. Tried to allocate 840.00 MiB (GPU 0; 14.56 GiB total capacity; 2.99 GiB already allocated; 237.50 MiB free; 4.24 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation.
展开
收起
1
条回答
写回答
写回答
-
您先试下torch.cuda.empty_cache()。可以参考自定义数据集
https://github.com/modelscope/swift/blob/main/docs/source/LLM/%E8%87%AA%E5%AE%9A%E4%B9%89%E4%B8%8E%E6%8B%93%E5%B1%95.md 此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”
2024-01-16 21:52:31赞同 展开评论 打赏
相关问答
热门讨论
热门文章
ModelScope中，模型下载默认路径在哪个路径？
9556
com/action/joingroup?code=v1是什么意思
7017
我希望通过damo-YOLO训练1500*1500的图片
8605
服务器上onnxruntime-gpu 调用结束，如何释放显存
5173
请问在 ModelScope 上的模型断网使用报这个错误啥原因了?
2173
ModelScope中我尝试使用这个模型搭建了一个推理服务，然后报错了怎么办？
3440
modelscope cache 环境变量是啥？
2716
请问modelscope中，有一个依赖，kantts，要怎么安装？
1949
streamlit 生成链接无法访问，gradio只生成内网，无法生成公网
2630
ModelScope中，请问Qwen-14B-Chat-Int4运行最少需要多少内存？
1781
展开全部
