翻小说 > 学霸的征途是星辰大海 > 第136章 计算中心

第136章 计算中心

    和室友们在食堂吃过早饭,徐辰便回到了自己的专属研究室。

    这一整天,他都把自己关在房间里,对着屏幕上的代码进行最后的精修。

    优化后的smooth_logic.py模块,代码量其实并不大,核心逻辑也就几百行。但其中涉及的Gumbel分布采样丶流形映射以及极其复杂的自定义梯度计算,对计算资源的要求却是指数级上升的。

    本书由??????????.??????全网首发

    当徐辰按下运行键的那一刻,不出意外的,爆显存了。

    徐辰无奈地停止了运行。

    之前那个简陋的Demo,用笔记本还能勉强跑一跑。

    但现在这个引入了高维几何嵌入的完全体模型,参数量虽然不如GPT那麽恐怖,但计算密度极高,显然不是一台笔记本能扛得住的。

    ……

    买工作站来不及,租云伺服器又担心代码泄露。徐辰想了想,决定找组织。

    他拨通了数院辅导员林轩的电话。

    如今徐辰已是数院的「重点保护对象」,享受讲师级待遇,林轩对他自然是格外上心,电话几乎是秒接,语气里透着股亲热劲儿。

    徐辰也没兜圈子,简单寒暄两句后,便道出了借用GPU伺服器的需求。

    不过结果有些遗憾。正如林轩所言,数院的超算集群设计之初就是为了服务于数值计算和代数推导,主打的是CPU核数和内存,GPU资源相当稀缺。仅有的几台带A100显卡的机器,也早已被应用数学系的老师们占满,用来跑流体力学方程,队都排到了下月了。

    但林轩很快指了条明路——校级计算中心。

    那是北大的公共大平台,硬体设施全是顶配,且遵循全校共享原则。相比于学院自己的「小灶」,那里才是真正的「大食堂」。只要去得早,大概率能申请到空闲节点。

    谢过林轩后,徐辰没再耽搁,背上书包便直奔理科楼群深处的计算中心。

    ……

    挂了电话,徐辰收拾好硬碟,直奔位于理科一号楼地下的校级计算中心。

    这里是北大科研算力的心脏。

    徐辰来到前台服务窗口。

    「你好,我想申请使用高性能GPU集群。」

    值班的老师是个戴眼镜的中年人,看了一眼徐辰的学生卡,又在系统里查了查。

    「运气不错,HPC-G分区刚好空出来一个A100节点,四个卡。前面的任务刚跑完。」老师推过来一张申请单,「扫个码,然后填一下这张表,签个字就行。」

    徐辰心中一喜。

    四张A100,这算力足够他在出发去德国前把模型训练完了。

    他拿起笔,刚写下自己的名字和院系。

    ……

    就在这时,大厅的自动门「哗啦」一声开了。

    一阵急促的脚步声传来。

    「快快快!老刘!江湖救急!」

    未见其人,先闻其声。

    紧接着,一队人火急火燎地冲到了窗口前。

    领头的是个穿着格子衬衫丶头发有些乱的男生,看样子也是个博士生或者青年教师,满头大汗,一脸焦急。手里还抱着一台笔记本电脑。身后跟着两个更年轻的学生,也是一脸的紧张。

    前台的中年老师抬起头,显然跟来人很熟:「哟,这不是物院的张博吗?怎麽急成这样?」

    「别提了!」被称作张博的男生擦了把汗,「我们那个『夸克-胶子等离子体』的模拟实验,参数出了点问题,得重新跑。但是下周就要在那边的《PhysicalReviewLetters》截稿了,这几天必须把数据跑出来!」

    「老刘,江湖救急,给我搞一组GPU节点,越快越好!」

    窗口里的张老师愣了一下,指了指正在填表的徐辰。

    「那个……实在不巧。刚才确实空出来一个节点,但是这位数院的同学……正在申请。」

    「什麽?!」

    那个被称为张博的男生猛地转过头,看向徐辰。

    当看到徐辰手里那张已经填了一半的申请单时,他的表情瞬间变得无比纠结,就像是被人抢走了最后一碗红烧肉的饿汉。

    他深吸了一口气,快步走到徐辰身边。

    ……

    虽然很急,但他并没有表现出蛮横,反而态度非常客气,甚至带着一丝恳求。

    「这位同学,你好。我是物理学院的博三学生,张乐阳。实在不好意思打扰你一下。」

    「是这样,我们团队有个非常紧急的物理模型需要跑,关系到一篇顶刊论文的截稿,时间非常紧迫。我看你也是刚来申请……」

    对方稍微顿了顿,然后试探着问道:

    「能不能麻烦你……把这个节点先让给我们?我们可以补偿你一些机时费,或者等我们跑完了,把我们的帐号借你用几天,你看行吗?」

    徐辰停下笔,看着眼前这位满头大汗丶眼神中透着渴望的师兄。

    他能理解这种为了科研deadline拼命的感觉。

    但是。

    这三周对他来说,同样珍贵。他也急着要在去德国之前,把AI的第一个版本叠代完。

    「师兄,不好意思。」

    「我的任务……也很急。」

    ……

    徐辰的拒绝让张乐阳愣了一下,刚想说什麽,手里的电话又响了。

    「喂,李老师……是,我在计算中心……什麽?隔壁震旦已经把预印本挂到ArXiv上了?数据跟我们预测的差不多?……好,好,我知道了!我这就跟他们协调,必须马上跑!」

    挂了电话,张乐阳的脸色变得极其难看,额头上的汗珠更密了,仿佛刚从水里捞出来一样。他看了一眼徐辰,眼神中多了一丝决绝,同时也带着深深的歉意。

    「同学,实在对不住了。」

    张乐阳叹了口气,转身走到角落,掏出手机拨通了一个号码。

    「喂,王主任吗?是我,小张……对,还是那个项目。我们这边急需一组算力,但是现在排满了……对,事发突然,复旦那边昨天发了预印本,数据跟我们要做的很像,老板急了,下了死命令必须今晚出结果验证……好好好,谢谢主任!改天让我老板请您喝茶!」

    挂断电话,不到两分钟,柜台里的工作人员便接到了内线电话。

    「嗯……好的,明白了。」

    工作人员放下电话,有些歉意地看向徐辰:「徐辰同学是吧?实在不好意思,中心刚接到通知,有一项紧急的国家重点研发计划任务需要优先保障。这组GPU节点……得先划拨给物理学院这边了。」

    「同学,真的对不住。」

    ……

    这就是行政力量。

    虽然不爽,但也无可厚非。

    在高校里,科研资源的分配本来就是极其残酷的。

    国家级重大专项>重点基金项目>一般项目>个人申请。这是写在潜规则里的铁律。

    徐辰皱了皱眉。

    虽然理智上能理解科研竞争的残酷性——那是真的争分夺秒,晚发一天,几个月的成果可能就打水漂了——但被人硬生生把到嘴边的肉抢走,心里多少还是有点不爽。

    ……

    张乐阳走过来,脸上满是愧疚,并没有那种「仗势欺人」的嚣张,反而显得很局促。

    「同学,真的抱歉。」张乐阳转过身,并没有那种「我抢了你资源理所当然」的傲慢,反而态度依旧放得很低,眼神里满是诚恳。

    「我也知道这事儿办得不地道,但我们那边真的火烧眉毛了。这样,这次你的机时费,全部算在我们课题组帐上,回头我让财务跟数院对接,绝不占用你的一分钱额度。」

    说着,他掏出手机,调出了自己的微信二维码,递到徐辰面前:

    「另外,咱们加个微信。这几天我们跑完数据,我把我们组这个优先级最高的队列借你用一周。」

    「我看你刚才准备提交的是个人申请,这种申请的优先级是最低的,就算之后你排到了,还是很有可能被挤掉的。」

    「算是我张乐阳欠你个人情,以后在北大,凡是跟物理学院有关的事,只要不违规,你随时来找我。」

    徐辰原本心里确实有些不爽,但是对方态度诚恳,给的条件也确实不错,自己就算没有被他们抢,之后也可能会被其他人抢,毕竟个人项目优先级实在太低了。心情好了一些,眉头舒展了一些。

    「算了。」徐辰叹了口气,拿出手机扫了一下二维码加了好友,「大家都是做科研的,都不容易。既然是国家重点项目,又是为了跟隔壁抢首发,你们先跑吧。」

    他不是那种不通情达理的人。而且看张乐阳那副快要急哭的样子,显然是被学术进度逼到了墙角,都是科研狗,相煎何太急。

    「谢了!太感谢了!以后有事说话!」张乐阳通过了好友申请,双手合十,连连作揖。

    ……