AI大牛李沐装机视频来了!你也能练100亿的大模型

人工智能 新闻
AI大牛沐神来装机了,还是训练100亿参数模型那种。

图片

在还没出装机视频前,李沐老师曾发起了一个小小的问卷调查,趁着显卡降价,看下童鞋们对装机跑Transformer有多大兴趣。

当时,就连华为天才少年「稚晖君」都来点赞了,足见大家还是很期待的。

图片

这不,沐神带着他的装机视频来了。怎样用最低的成本训练一个100亿模型?

而就在最近,币圈也在一直降温,同时GPU也明显降价了不少,就比如英伟达3090TI现在售价1600美元(原价2000美元)。

图片

在这个项目的开始,先装了2台原型机,每台机器都是双卡RTX 3090TI,采用了水冷散热系统,减少了噪音。

一台机器装下来,造价有5000多美元,大约3万5人民币。

话不多说,来看看沐神如何装机的吧~

装机清单

装机的需求,首先就要是足够安静。不然太吵的话没法工作。

第二个需求就是散热要好。不然温度过高的话会导致GPU降频。

第三个,重点来了,因为需要跑比较大的Transformer模型,所以GPU的带宽必须足够好。

如果之前装过GPU服务器用来跑CNN的话,跟跑Transformer模型的需求会不太一样。因为Transformer模型比CNN模型要更大,对内存的占有率会更高。所以GPU的内存大小非常重要。

图片

之前沐神也讲过,要把这样大的Transformer模型放到多个GPU上训练,来自谷歌、微软等的工程师都是使用类似DGA X100这样的机器来跑的。即使在这样的机器上,GPU的带宽仍然是一个瓶颈。

购买这种服务器GPU和游戏GPU的区别就在于,前者不在于单卡能跑多快,而是卡与卡之间能够多快地连接出来。

因此装机理念的重点就是:尽量增大GPU内存,以及GPU之间互联的一个带宽,

如果一台机器想放很多卡的话,就要买涡轮的散热。

如果想要安静的话,就买水冷的散热,沐神买的是4块3090 TI。用水冷的好处是比较安静,坏处就是特别占地方。

所以,如果要在机箱放四块卡的话,就不要买水冷的版本,而是要买只有一个涡轮风扇的版本。

而且机箱里风的流向是一个特别重要的问题。如果买带3个风扇的卡,风是从正面进入机箱,然后从四面八方散热,卡挨得紧的话,机箱里的温度就会非常高。

沐神也说,自己在多年前,买了四块2个大风扇的卡放在一起,结果就导致一块卡温度过高烧掉了。

GPU选好以后(华硕的ROG), 剩下的配置就比较简单。CPU用的是AMD的12核CPU,主板是号称PCIE 4.0 16的某牌子,硬盘是2 TB的M.2的硬盘,风扇是120毫米水冷风扇,加一个全尺寸机箱。

图片

装机步骤

装机清单完成后,接下来就是具体的装机过程了。步骤如下:

首先放GPU。注意,放的过程中一定不能用手碰金属的地方,如果有静电的话非常容易造成GPU导电。

图片

把GPU放进去之后,把螺丝拧上。然后把风扇装进去。

插上电源后,把电源线和水管线绑在一起。然后把NVLink的桥给连上去。

最后连上电源,机器就可以运行了。

图片

压力测试

在装机完成后,下一步的任务就是继续装上操作系统。

沐神装的是ubuntu22,装上之后就靠远程连上去了。

当然了,沐神也是把各种情况说的比较详细,除了ubuntu22以外,windows和linux在不同的需求下也是可以的。

这里沐神用的SSH进行的远程连接。

图片

沐神的系统已经装上了驱动,同时,他也指出如果还没有驱动的话,也可以用apt-get装上nvidia-driver-515。

图片

装好之后,就可以运行nvidia-smi,看到系统了。

图片

从中可以看到各项信息。比如GPU的数量、温度、瓦数、内存使用等等。

接下来还可以通过nvidia-smi的topo-m矩阵看到nv-link是否正常。

图片

可以看到两个GPU由NV4连接。4表示4个通道,这就代表连接是正常的。

下一题,是测试系统在满负荷的情况下的温度。

沐神表示,测试GPU的是一个叫gpu-burn的小程序,github上可以下载。

这里沐神模拟跑了十分钟,也是看到了两个GPU的温度。沐神还打趣说,都能感觉到GPU在呼呼吹热风。

同理,CPU也可以用这种办法测试温度,用的是cpu-burn。

最终两个GPU的温度停留在58度和55度,功耗拉到了440多瓦(满功耗480瓦),还挺好的。

最后的一项参数是机器的耗电情况。沐神的测试大概用了1240瓦,意味着每小时用电1.5度。

图片

从目前的数据来看,稳定性还ok。

至于用这台机器跑Transformer性能怎么样,还得等下期视频了。

网友热评

视频发出后,b站的网友也表示出了极大的兴趣。

有满分课代表同学出没,列出了视频中提到的完整配置清单。图片

还有网友火速前来围观,「跟李沐学装机」。

图片


沐神自己表示,感觉3090ti的卡不是太行。马上有网友搭腔,「不行就抽了送人吧。」

图片

当然了,这种硬核装机视频下的评论肯定少不了幽默因素。

图片

只能说,太真实了。

责任编辑:张燕妮 来源: 新智元
相关推荐

2023-03-07 18:37:20

ChatGPTAI

2021-05-26 17:33:47

AI 数据人工智能

2024-01-15 16:34:13

模型训练

2023-03-08 14:36:38

AI模型

2023-11-26 17:03:33

公司AI

2023-11-06 09:26:00

模型测评

2023-05-16 12:11:22

2022-12-27 16:31:22

AI绘图软件

2023-04-21 10:14:22

开源AI

2023-06-26 07:51:48

2021-05-06 09:17:05

AI 数据人工智能

2021-03-26 10:02:29

PythonVIP视频看电影

2021-10-11 17:30:49

AI 人脸识别人工智能

2024-04-19 13:53:20

2023-05-10 14:40:40

AI模型算力

2024-01-30 13:02:05

AI训练
点赞
收藏

51CTO技术栈公众号