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

图片

相关推荐
花椒技术1 小时前
AI 协同开发落地复盘:1 小时生成首版后,为什么 Review 和修正又花了 2-3 天
前端·人工智能·架构
Gerardisite1 小时前
CRM、ERP、OA 如何连接企业微信?QiWe 提供标准化解决方案
java·python·机器人·自动化·企业微信
城管不管1 小时前
Maven Helper
java·macos·intellij-idea
ch.ju1 小时前
Java程序设计(第3版)第三章——数组的动态获取
java·开发语言
Java知识技术分享2 小时前
策略模式的两种实现:抽象类和接口
java·spring·策略模式
液态不合群2 小时前
Redis--哨兵机制与CAP定理
java·redis·bootstrap
曹牧2 小时前
Java:PDF文件扁平化处理
java·开发语言·pdf
灰色人生qwer2 小时前
解决IDEA运行Java程序jdk版本不匹配问题
java·开发语言·intellij-idea
万少2 小时前
万少用9个AI工具,帮朋友完成了一个"不可能"的项目
前端