指定的ModelScope gpu，但是只用到了一张2卡，怎么修改？
"torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 28.57 GiB (GPU 2; 39.59 GiB total capacity; 33.25 GiB already allocated; 5.06 GiB free; 33.60 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF我用这种方式os.environ['CUDA_VISIBLE_DEVICES']='0,1,2,3,4,5,6,7' 指定的ModelScope gpu，但是只用到了一张2卡，怎么修改？
"
展开
收起
1
条回答
写回答
写回答
-
"参考下mp+ddp，https://github.com/modelscope/swift/blob/main/examples/pytorch/llm/scripts/dpo/lora_ddp_mp/dpo.sh
此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”"
2024-03-12 20:28:34赞同 展开评论 打赏
相关问答
ModelScope旨在打造下一代开源的模型即服务共享平台，为泛AI开发者提供灵活、易用、低成本的一站式模型服务产品，让模型应用更简单！欢迎加入技术交流群：微信公众号：魔搭ModelScope社区，钉钉群号：44837352
热门讨论
热门文章
ModelScope qwen7b、14b、72b lora微调分别需要的显存和cpu内存是多少？
4595
modelscope-funasr的FunClip怎么安装的？
1806
小白求助，跟着大模型微调教程做不太明白
719
ModelScope中，模型下载默认路径在哪个路径？
9561
com/action/joingroup?code=v1是什么意思
7034
ModelScope默认的下载位置是在哪里？
2568
ModelScope有部署qwen2-72b的么，正常推理需要多大的显存？
3676
我希望通过damo-YOLO训练1500*1500的图片
8611
无法导入 modelscope，已经pip install了
3310
服务器上onnxruntime-gpu 调用结束，如何释放显存
5177
展开全部
阿里云通义千问向全社会开放！
39433
智胜未来：国内大模型+Agent应用案例精选，以及主流Agent框架开源项目推荐
1554
multi-agent：多角色Agent协同合作，高效完成复杂任务
32962
Llama 3开源，魔搭社区手把手带你推理，部署，微调和评估
11245
ModelScope一键部署模型：新手村实操FAQ篇
4902
通义千问72B、1.8B、Audio模型开源！魔搭社区最佳实践来啦！
5465
大模型时代，还缺一只雨燕 ｜ SWIFT：魔搭社区轻量级微调推理框架
3895
自然语言处理Transformer模型最详细讲解（图解版）
6294
“魔搭”来了！一文深度解读达摩院推出的AI模型社区
7214
TripoSR开源！从单个图像快速生成 3D 对象！（附魔搭社区推理实战教程）
1530
展开全部
