DeePMD-kit一键部署 | 优云智算

DeepMD-kit 是一个基于深度学习的分子动力学模拟软件包，旨在生成具有量子力学精度的势能面（PES）。 本项目环境配置使用 PyTorch 作为计算后端，针对 NVIDIA RTX 4090 (Ada Lovelace 架构) 和 CUDA 12.8 进行了适配。
为了避免 DeepMD-kit 错误地调用 TensorFlow，必须设置以下环境变量：
建议将上述命令加入 ~/.bashrc 或 Conda 的激活脚本中。
使用官方提供的水分子（Water）案例来验证 GPU 是否正常介入训练。
或者，如果您已经设置了环境变量 DP_BACKEND=pytorch：
1. 检查日志输出 终端应显示类似以下的训练进度，包含 batch、l2_err 等信息：
2. 检查 GPU 占用 在训练过程中，打开一个新的终端窗口，输入：
请观察 python 进程是否出现在列表底部，且 GPU-Util（利用率）有明显浮动（非 0%），这证明 RTX 4090 正在工作。
Q: 出现 ImportError: cannot import name 'main' from 'deepmd.tf...' 错误？
Q: 出现 torch not compiled with CUDA enabled？
隶属于优刻得科技股份有限公司 | 优刻得是中立、安全的云计算服务平台