指定的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
热门讨论
热门文章
通义千问2.5-7B-Instruct已经下载到本地为什么使还需要联网？而且最近下载模型也提示400
434
ModelScope中，模型下载默认路径在哪个路径？
9937
modelscope-funasr的FunClip怎么安装的？
2168
CUDA error: CUDA-capable device(s) is busy
320
modelscope下载 的模型 如何转ollama
198
ModelScope qwen7b、14b、72b lora微调分别需要的显存和cpu内存是多少？
5052
com/action/joingroup?code=v1是什么意思
7413
ModelScope有部署qwen2-72b的么，正常推理需要多大的显存？
3942
请问在 ModelScope 上的模型断网使用报这个错误啥原因了?
2335
我希望通过damo-YOLO训练1500*1500的图片
8795
展开全部
阿里云通义千问向全社会开放！
48099
multi-agent：多角色Agent协同合作，高效完成复杂任务
33571
Llama 3开源，魔搭社区手把手带你推理，部署，微调和评估
11819
智胜未来：国内大模型+Agent应用案例精选，以及主流Agent框架开源项目推荐
2004
自然语言处理Transformer模型最详细讲解（图解版）
6638
通义千问72B、1.8B、Audio模型开源！魔搭社区最佳实践来啦！
5768
vanna+qwen实现私有模型的SQL转换
5086
开源版图生视频I2VGen-XL：单张图片生成高质量视频
4009
Sketch2Lineart：AI绘画工具，自动将手绘草图转换成清晰的线条画
77
从RMSE和MSE到更多选择：探索机器学习模型性能指标
3843
展开全部
