指定的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
热门讨论
热门文章
最近大模型降价潮，哪家才是真的“价美”也“物美”？
2767
com/action/joingroup?code=v1是什么意思
3945
ModelScope中，模型下载默认路径在哪个路径？
3039
ModelScope qwen7b、14b、72b lora微调分别需要的显存和cpu内存是多少？
1173
无法导入 modelscope，已经pip install了
1346
如何训练属于自己的“通义千问”呢？
5151
ModelScope 使用14B模型推理时报这个错误？
984
我希望通过damo-YOLO训练1500*1500的图片
7149
docker和open-webui、ollama 部署本地的qwen2 14b 提供了API接口吗？
70
请问ModelScope中，怎么修改本地运行，模型自动下载到 c 盘下？
3141
展开全部
Llama 3开源！魔搭社区手把手带你推理，部署，微调和评估
5448
阿里云通义千问向全社会开放！
20895
Phi-3：小模型，大未来！（附魔搭社区推理、微调实战教程）
1367
vanna+qwen实现私有模型的SQL转换
1588
multi-agent：多角色Agent协同合作，高效完成复杂任务
27166
ModelScope--人像卡通化、人像美肤
52497
ChatGPT中文版杀疯了,已登录AI模型市场
12211
万字长文 | 多目标跟踪最新综述（基于Transformer/图模型/检测和关联/孪生网络）（上）
1709
自然语言处理Transformer模型最详细讲解（图解版）
4102
大模型时代，还缺一只雨燕 ｜ SWIFT：魔搭社区轻量级微调推理框架
1849
展开全部
