AMSET 设置多核并行计算

AMSET是基于形变势理论的自动化输运计算工具,能够从 VASP 能带输出出发,一站式给出载流子迁移率、电导率、Seebeck 系数等关键输运性质。

随着掺杂浓度和温度点的增多,散射率计算量急剧膨胀,多核并行成为缩短机时的必备手段。

AMSET 内部采用多进程并行 来加速散射率计算,默认情况下会尝试使用全部可用的 CPU 核心(nworkers = -1)。

但这在集群或个人工作站上很容易触发内存耗尽问题,导致进程被杀。

方式一:

settings.yaml 中,通过 nworkers 控制并发进程数:

nworkers: 4 # 指定使用 4 个进程

方式二:

在启动 amset run 前设置环境变量后,直接加入并行参数

export OMP_NUM_THREADS=1 # 限制每个进程内部只开1个线程

export MKL_NUM_THREADS=1 # 如果使用 Intel MKL

export OPENBLAS_NUM_THREADS=1 # 如果后端是 OpenBLAS

amset run --nworkers 24

图片

相关推荐
mCell6 分钟前
用 Knip 清理 AI Coding 留下的冗余代码
前端·ai编程·前端工程化
鹿角片ljp16 分钟前
KV Cache 解析
java·算法
captain37632 分钟前
网络原理(9)-数据链路层
java·网络协议·java-ee
笃行3501 小时前
使用Rokid AIUI做了一个童年推箱子游戏
前端
惊讶的猫1 小时前
意图识别置信度和检索重排的分数
java
caoerzhong1 小时前
仓库管理数字化趋势解读:JeeWMS 开源 Java WMS 视角下的五个演进方向
java·开源
程序员JerrySUN2 小时前
Jetson Edge AI 实战01:Nano、Xavier、Orin 怎么选?Jetson 硬件选型详解【视频讲解】
java·数据库·redis·安全·mybatis
excel2 小时前
前端加密的作用与使用场景
前端
Joy T3 小时前
Spring AI 2.0 Agent 进阶:Memory、State 与 Context Engineering 常见技术全景
java·人工智能·后端·spring·agent入门·agent state