智能体使用经验 GPU4PySCF 超内存问题

mhh · 2025年05月06日 · 64 次阅读

当前平台 (version=0.1.0) 部署的 GPU4PySCF,计算资源的调用在两个地方控制:(1) 用户不可更改:每个 GPU4PySCF 任务申请两块 GPU,约 50 G 内存,另外申请 32 个 CPU 核,每个 CPU 核一般有 4G≈4000 MB 内存;(2) 用户可更改:计算条件配置文件(比如 gpu4pyscf_tddft_config.json),默认设置线程数为 8,最大内存数为 32000 MB(这个是指 CPU 的内存),如下所示:

用户可修改这两个参数,按照 1:4 的配比来修改,比如

修改方式:既可以让 Agent 修改,也可由用户编辑修改。

最后,关于 GPU 使用的部分,应该是两块 GPU 的全部内存,之后会和 GPU4PySCF 的开发者询问,完善本贴的讨论。日后,可能根据朋友们的反馈,以及考虑其他因素,确定是否由用户还是 Agent 来决定计算资源。

暂无回复。
需要 登录 后方可回复, 如果你还没有账号请 注册新账号