第136章 计算中心
和室友们在食堂吃过早饭,徐辰便回到了自己的专属研究室。
这一整天,他都把自己关在房间里,对着屏幕上的代码进行最后的精修。
优化后的 smooth_logic.py模块,代码量其实并不大,核心逻辑也就几百行。但其中涉及的Gumbel分布采样、流形映射以及极其复杂的自定义梯度计算,对计算资源的要求却是指数级上升的。
当徐辰按下运行键的那一刻,不出意外的,爆内存(Mac内存显存一体)了。
徐辰无奈地停止了运行。
之前那个简陋的Demo,用笔记本还能勉强跑一跑。
但现在这个引入了高维几何嵌入的完全体模型,参数量虽然不如GPT那么恐怖,但计算密度极高,显然不是一台笔记本能扛得住的。
……
买工作站来不及,租云服务器又担心代码泄露。徐辰想了想,决定找组织。
他拨通了数院辅导员林轩的电话。
如今徐辰已是数院的“重点保护对象”,享受讲师级待遇,林轩对他自然是格外上心,电话几乎是秒接,语气里透着股亲热劲儿。
徐辰也没兜圈子,简单寒暄两句后,便道出了借用GPU服务器的需求。
不过结果有些遗憾。正如林轩所言,数院的超算集群设计之初就是为了服务于数值计算和代数推导,主打的是CPU核数和内存,GPU资源相当稀缺。仅有的几台带A100显卡的机器,也早已被应用数学系的老师们占满,用来跑流体力学方程,队都排到了下月了。
但林轩很快指了条明路——校级计算中心。
那是北大的公共大平台,硬件设施全是顶配,且遵循全校共享原则。相比于学院自己的“小灶”,那里才是真正的“大食堂”。只要去得早,大概率能申请到空闲节点。
谢过林轩后,徐辰没再耽搁,背上书包便直奔理科楼群深处的计算中心。
……
挂了电话,徐辰收拾好硬盘,直奔位于理科一号楼地下的校级计算中心。
这里是北大科研算力的心脏。
徐辰来到前台服务窗口。
“你好,我想申请使用高性能GPU集群。”
值班的老师是个戴眼镜的中年人,看了一眼徐辰的学生卡,又在系统里查了查。
“运气不错,HPC-G分区刚好空出来一个A100节点,四个卡。前面的任务刚跑完。”老师推过来一张申请单,“扫个码,然后填一下这张表,签个字就行。”
徐辰心中一喜。
四张A100,这算力足够他在出发去德国前把模型训练完了。
他拿起笔,刚写下自己的名字和院系。
……
就在这时,大厅的自动门“哗啦”一声开了。
一阵急促的脚步声传来。
“快快快!老刘!江湖救急!”
未见其人,先闻其声。
紧接着,一队人火急火燎地冲到了窗口前。
领头的是个穿着格子衬衫、头发有些乱的男生,看样子也是个博士生或者青年教师,满头大汗,一脸焦急。手里还抱着一台笔记本电脑。身后跟着两个更年轻的学生,也是一脸的紧张。
前台的中年老师抬起头,显然跟来人很熟:“哟,这不是物院的张博吗?怎么急成这样?”
“别提了!”被称作张博的男生擦了把汗,“我们那个‘夸克-胶子等离子体’的模拟实验,参数出了点问题,得重新跑。但是下周就要在那边的《Physical Review Letters》截稿了,这几天必须把数据跑出来!”
“老刘,江湖救急,给我搞一组GPU节点,越快越好!”
窗口里的张老师愣了一下,指了指正在填表的徐辰。
“那个……实在不巧。刚才确实空出来一个节点,但是这位数院的同学……正在申请。”
“什么?!”
那个被称为张博的男生猛地转过头,看向徐辰。
当看到徐辰手里那张已经填了一半的申请单时,他的表情瞬间变得无比纠结,就像是被人抢走了最后一碗红烧肉的饿汉。
他深吸了一口气,快步走到徐辰身边。
……
虽然很急,但他并没有表现出蛮横,反而态度非常客气,甚至带着一丝恳求。
“这位同学,你好。我是物理学院的博三学生,张乐阳。实在不好意思打扰你一下。”
“是这样,我们团队有个非常紧急的物理模型需要跑,关系到一篇顶刊论文的截稿,时间非常紧迫。我看你也是刚来申请……”
对方稍微顿了顿,然后试探着问道:
“能不能麻烦你……把这个节点先让给我们?我们可以补偿你一些机时费,或者等我们跑完了,把我们的账号借你用几天,你看行吗?”
徐辰停下笔,看着眼前这位满头大汗、眼神中透着渴望的师兄。
他能理解这种为了科研deadline拼命的感觉。
但是。
这三周对他来说,同样珍贵。他也急着要在去德国之前,把AI的第一个版本迭代完。
“师兄,不好意思。”
“我的任务……也很急。”
……
徐辰的拒绝让张乐阳愣了一下,刚想说什么,手里的电话又响了。
“喂,李老师……是,我在计算中心……什么?隔壁震旦已经把预印本挂到ArXiv上了?数据跟我们预测的差不多?……好,好,我知道了!我这就跟他们协调,必须马上跑!”
挂了电话,张乐阳的脸色变得极其难看,额头上的汗珠更密了,仿佛刚从水里捞出来一样。他看了一眼徐辰,眼神中多了一丝决绝,同时也带着深深的歉意。
“同学,实在对不住了。”
张乐阳叹了口气,转身走到角落,掏出手机拨通了一个号码。
“喂,王主任吗?是我,小张……对,还是那个项目。我们这边急需一组算力,但是现在排满了……对,事发突然,复旦那边昨天发了预印本,数据跟我们要做的很像,老板急了,下了死命令必须今晚出结果验证……好好好,谢谢主任!改天让我老板请您喝茶!”
挂断电话,不到两分钟,柜台里的工作人员便接到了内线电话。
“嗯……好的,明白了。”
工作人员放下电话,有些歉意地看向徐辰:“徐辰同学是吧?实在不好意思,中心刚接到通知,有一项紧急的国家重点研发计划任务需要优先保障。这组GPU节点……得先划拨给物理学院这边了。”
“同学,真的对不住。”
……
这就是行政力量。
虽然不爽,但也无可厚非。
在高校里,科研资源的分配本来就是极其残酷的。
国家级重大专项>重点基金项目>一般项目>个人申请。这是写在潜规则里的铁律。
徐辰皱了皱眉。
虽然理智上能理解科研竞争的残酷性——那是真的争分夺秒,晚发一天,几个月的成果可能就打水漂了——但被人硬生生把到嘴边的肉抢走,心里多少还是有点不爽。
……
张乐阳走过来,脸上满是愧疚,并没有那种“仗势欺人”的嚣张,反而显得很局促。
“同学,真的抱歉。”张乐阳转过身,并没有那种“我抢了你资源理所当然”的傲慢,反而态度依旧放得很低,眼神里满是诚恳。
“我也知道这事儿办得不地道,但我们那边真的火烧眉毛了。这样,这次你的机时费,全部算在我们课题组账上,回头我让财务跟数院对接,绝不占用你的一分钱额度。”
说着,他掏出手机,调出了自己的微信二维码,递到徐辰面前:
“另外,咱们加个微信。这几天我们跑完数据,我把我们组这个优先级最高的队列借你用一周。”
“我看你刚才准备提交的是个人申请,这种申请的优先级是最低的,就算之后你排到了,还是很有可能被挤掉的。”
“算是我张乐阳欠你个人情,以后在北大,凡是跟物理学院有关的事,只要不违规,你随时来找我。”
徐辰原本心里确实有些不爽,但是对方态度诚恳,给的条件也确实不错,自己就算没有被他们抢,之后也可能会被其他人抢,毕竟个人项目优先级实在太低了。心情好了一些,眉头舒展了一些。
“算了。”徐辰叹了口气,拿出手机扫了一下二维码加了好友,“大家都是做科研的,都不容易。既然是国家重点项目,又是为了跟隔壁抢首发,你们先跑吧。”
他不是那种不通情达理的人。而且看张乐阳那副快要急哭的样子,显然是被学术进度逼到了墙角,都是科研狗,相煎何太急。
“谢了!太感谢了!以后有事说话!”张乐阳通过了好友申请,双手合十,连连作揖。
……
(https://www.piaotian55.com/book/798348704/40924032.html)
1秒记住飘天文学网:www.piaotian55.com。手机版阅读网址:m.piaotian55.com