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