水木社区手机版
首页
|版面-人工智能(AI)|
新版wap站已上线
返回
1/1
|
转到
主题:一直不太理解,为什么训练大模型不能利用SSD
33楼
|
StephenLee
|
2023-11-11 08:39:31
|
展开
别一上来就说没有,有框架支持 offload 到本地磁盘呀,deepspeed 可以。一般用 NVMe 盘。
【 在 Xjt (Voldemort) 的大作中提到: 】
: 一直不太理解,为什么训练大模型不能利用SSD
:
: 现在制约大模型训练和推理的,根本不是芯片速度,而是内存。想要全量微调一个6B的大模型,就得有320G显存,就得4块A100。
:
--
FROM 125.119.104.*
1/1
|
转到
选择讨论区
首页
|
分区
|
热推
BYR-Team
©
2010.
KBS Dev-Team
©
2011
登录完整版