项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-09-16 · 当前星标 ★ 3417
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- Python 环境(具体支持的版本以仓库 pyproject.toml 为准)
- 可选 NVIDIA CUDA 环境:README 提供 cu12 与 cu13 两种后端
- 使用 pip 安装包,或使用 uv 从源码同步依赖
安装与启动步骤
-
1确认 Python 版本
README 未写明具体版本号,请先查看仓库 pyproject.toml 中当前支持的 Python 版本、可选依赖组与依赖约束。
-
2安装 CUDA 13 后端
README 顶部命令默认选择 CUDA 13 后端,CUDA 13 环境直接安装该后端。
pip install "nvidia-physicsnemo[cu13]" -
3改用 CUDA 12 后端
若本机为 CUDA 12 环境,把后端换成 cu12 即可,其余用法一致。
pip install "nvidia-physicsnemo[cu12]" -
4基础安装
基础安装使用 PyPI 默认的 PyTorch 发行版,不安装 CUDA 专用的 RAPIDS 包。
pip install nvidia-physicsnemo -
5叠加可选功能
可选功能组可与任一后端组合,方括号内用逗号分隔,例如同时启用 gnns。
pip install "nvidia-physicsnemo[cu13,gnns]" -
6源码方式安装
从 GitHub 检出源码,用 uv 同步依赖,--extra 指定后端。
git clone https://github.com/NVIDIA/physicsnemo.git cd physicsnemo uv sync --extra cu13
如何确认成功
README 未给出验证命令;可执行 pip show nvidia-physicsnemo 确认包已安装成功。
常见问题
Q:应该选 cu12 还是 cu13 后端?
A:取决于本机 CUDA 版本。README 顶部默认命令对应 CUDA 13 后端;CUDA 12 环境把后端换成 cu12,命令结构完全相同。
Q:不装 CUDA 相关的 RAPIDS 包可以吗?
A:可以。直接执行 pip install nvidia-physicsnemo 即为基础安装,使用 PyPI 默认的 PyTorch 发行版。
Q:如何同时启用多个可选功能?
A:在方括号内用逗号组合后端与功能组,例如 pip install "nvidia-physicsnemo[cu13,gnns]"。
Q:支持哪些 Python 版本?
A:README 未直接列出,请查看仓库中的 pyproject.toml,其中给出了支持的 Python 版本与精确依赖约束。
注意事项
- 依赖版本、支持范围与可选依赖组以仓库 pyproject.toml 为准
- cu12 与 cu13 后端任选其一,可选功能组可与任一后端叠加
- 源码方式安装需要 uv 工具,请先按 uv 文档安装
核心亮点
- 原生支持 PINN、FNO、DeepONet、GNN 等主流 Physics-ML 模型,开箱即用
- 深度集成 NVIDIA Warp 与 CUDA,可微分物理求解与 GPU 加速训练一体化
- 由 NVIDIA 维护,文档、示例与预训练模型较完整,工程化程度高
不足之处
- 强依赖 NVIDIA GPU 生态,非 N 卡或纯 CPU 环境体验受限
- 物理约束调参与损失权重设计门槛较高,新手学习曲线偏陡
- 版本迭代较快,部分 API 与旧版 Modulus 代码存在兼容迁移成本
适用场景
- 用 PINN 求解 CFD/传热偏微分方程,替代部分传统数值仿真
- 训练神经算子代理模型,实现实时参数化仿真与数字孪生
- 为油气、气象、电磁等工业场景构建物理约束的 AI 代理模型
替代项目
DeepXDE、Modulus(旧版)、PhiFlow
同类项目推荐
LMFlow
开源
一键微调大模型,人人可用的高效工具包
An Extensible Toolkit for Finetuning and Inference of Large Foundation Models. Large···
SpecForge
开源
训练投机解码模型,一键部署到 SGLang 提速
Train speculative decoding models effortlessly and port them smoothly to SGLang serv···
AnglE
开源
用角度学习训练最强句子向量,登顶MTEB榜单
Train and Infer Powerful Sentence Embeddings with AnglE | SOTA on STS and MTEB Lead···
nnscaler
开源
写单卡模型,编译器自动生成多卡并行训练方案
nnScaler: Compiling DNN models for Parallel Training
PSEC
开源
让AI智能体像搭积木一样,在参数空间里不断学习新技能、组合旧技能,应对新挑战。
[ICLR 2025] The offical implementation of "PSEC: Skill Expansion and Composition in ···
quaterion
开源
几分钟搭好相似度模型训练,向量检索效果快速提升
Blazing fast framework for fine-tuning similarity learning models
Complete-LLM-Finetuning
开源
跟着 Notebook 跑一遍,就能亲手微调大模型
MLOps
开源
跟着项目清单,动手练熟MLOps全流程
The repository contains a list of projects which I will work on while learning and i···
wandb
开源
训练过程实时看曲线,实验记录自动归档不丢
The AI developer platform. Use Weights & Biases to train and fine-tune models, and m···
