当前位置: 首页 > news >正文

Accelerate单卡,多卡config文件配置

 依赖库

from accelerate import Accelerator
from accelerate import DistributedDataParallelKwargs
ddp_kwargs = DistributedDataParallelKwargs(find_unused_parameters=True)
accelerator = Accelerator(kwargs_handlers=[ddp_kwargs])

 代码中删除所有的.cuda() 或者to(device),损失函数的可能不需要删除。因为我删了报错了。。。具体原因目前未知。

单机单卡训练

单机多卡

单机单卡多卡切换

直接看这个大哥的博客就可以 

https://blog.csdn.net/zjc910997316/article/details/134642194?ops_request_misc=&request_id=&biz_id=102&utm_term=accelerate%20config%E5%A6%82%E4%BD%95%E6%97%A2%E5%8F%AF%E4%BB%A5%E5%8D%95%E5%8D%A1%E5%8F%88%E5%8F%AF%E4%BB%A5%E5%8F%8C%E5%8D%A1&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-0-134642194.142^v100^pc_search_result_base9&spm=1018.2226.3001.4187

 


http://www.mrgr.cn/news/38511.html

相关文章:

  • 对小白友好的与易我同级别四款剪辑工具推荐:
  • Xshell连接服务器
  • Spring Boot 整合 Keycloak
  • 花都狮岭寄宿自闭症学校:开启孩子的生命之门
  • 使用python获取百度一下,热搜TOP数据详情
  • 芝士AI写作有什么特色? 大模型支撑,智能改写续写,让写作更轻松
  • RHCS认证-Linux(RHel9)-Ansible
  • Python自学查漏9.28
  • [论文阅读] ChartInstruct: Instruction Tuning for Chart Comprehension and Reasoning
  • Java高效编程(10):重写equals时必须遵循通用约定
  • MySQL(面试题 - 同类型归纳面试题)
  • MySQL 之多表设计详解
  • 优雅使用 MapStruct 进行类复制
  • EDM平台大比拼 用户体验与营销效果双重测评
  • 使用jdframe进行数据处理
  • 初始爬虫9
  • Mybatis(三)
  • 信道衰落的公式
  • 【大模型】AutoDL部署AI绘图大模型Stable Diffusion使用详解
  • C#从入门到精通(30)—C#Marshal类用法总结