第136章 计算中心(1/2)
和室友们在食堂吃过早饭,徐辰便回到了自已的专属研究室。
这一整天,他都把自已关在房间里,对着屏幕上的代码进行最后的精修。
优化后的sooth_logic.y模块,代码量其实并不大,核心逻辑也就几百行。但其中涉及的Gubel分布采样、流形映射以及极其复杂的自定义梯度计算,对计算资源的要求却是指数级上升的。
当徐辰按下运行键的那一刻,不出意外的,爆内存(Mac内存显存一体)了。
徐辰无奈地停止了运行。
之前那个简陋的Deo,用笔记本还能勉强跑一跑。
但现在这个引入了高维几何嵌入的完全体模型,参数量虽然不如GPT那么恐怖,但计算密度极高,显然不是一台笔记本能扛得住的。
……
买工作站来不及,租云服务器又担心代码泄露。徐辰想了想,决定找组织。
他拨通了数院辅导员林逸的电话。
如今徐辰已是数院的“重点保护对象”,享受讲师级待遇,林逸对他自然是格外上心,电话几乎是秒接,语气里透着股亲热劲儿。
徐辰也没兜圈子,简单寒暄两句后,便道出了借用GPU服务器的需求。
不过结果有些遗憾。正如林逸所言,数院的超算集群设计之初就是为了服务于数值计算和代数推导,主打的是CPU核数和内存,GPU资源相当稀缺。仅有的几台带A100显卡的机器,也早已被应用数学系的老师们占满,用来跑流体力学方程,队都排到了下月了。
但林逸很快指了条明路——校级计算中心。
那是北大的公共大平台,硬件设施全是顶配,且遵循全校共享原则。相比于学院自已的“小灶”,那里才是真正的“大食堂”。只要去得早,大概率能申请到空闲节点。
谢过辅导员后,徐辰没再耽搁,背上书包便直奔理科楼群深处的计算中心。
……
挂了电话,徐辰收拾好硬盘,直奔位于理科一号楼地下的校级计算中心。
这里是北大科研算力的心脏。
徐辰来到前台服务窗口。
“你好,我想申请使用高性能GPU集群。”
值班的老师是个戴眼镜的中年人,看了一眼徐辰的学生卡,又在系统里查了查。
“运气不错,HPC-G分区刚好空出来一个A100节点,四个卡。前面的任务刚跑完。”老师推过来一张申请单,“扫个码,然后填一下这张表,签个字就行。”
徐辰心中一喜。
四张A100,这算力足够他在出发去德国前把模型训练完了。
他拿起笔,刚写下自已的名字和院系。
……
就在这时,大厅的自动门“哗啦”一声开了。
一阵急促的脚步声传来。
“快快快!老刘!江湖救急!”
未见其人,先闻其声。
紧接着,一队人火急火燎地冲到了窗口前。
领头的是个穿着格子衬衫、头发有些乱的男生,看样子也是个博士生或者青年教师,满头大汗,一脸焦急。手里还抱着一台笔记本电脑。身后跟着两个更年轻的学生,也是一脸的紧张。
前台的中年老师抬起头,显然跟来人很熟:“哟,这不是物院的张博吗?怎么急成这样?”
“别提了!”被称作张博的男生擦了把汗,“我们那个‘夸克-胶子等离子体’的模拟实验,参数出了点问题,得重新跑。但是下周就要在那边的《PhysicalReviewLetters》截稿了,这几天必须把数据跑出来!”
“老刘,江湖救急,给我搞一组GPU节点,越快越好!”
窗口里的张老师愣了一下,指了指正在填表的徐辰。
“那个……实在不巧。刚才确实空出来一个节点,但是这位数院的同学……正在申请。”
“什么?!”
那个被称为张博的男生猛地转过头,看向徐辰。
当看到徐辰手里那张已经填了一半的申请单时,他的表情瞬间变得无比纠结,就像是被人抢走了最后一碗红烧肉的饿汉。
他深吸了一口气,快步走到徐辰身边。
……
本章未完,点击下一页继续阅读。