这个目录用于学习 Physics-Informed Neural Networks,当前包含一个 PyTorch 手写 ODE PINN 入门示例,以及 PINN/DeepXDE 速查文档。
项目虚拟环境位于:
/home/guiyu/workspace/PINN/.venv已验证的主要依赖:
torch 2.12.0+cpu
DeepXDE 1.15.0
numpy 2.2.6
matplotlib 3.10.9
scipy 1.15.3
pandas 2.3.3
tqdm 4.68.1
PyTorch 是 CPU 版:
torch.version.cuda = None
torch.cuda.is_available() = False
cd /home/guiyu/workspace/PINN
source .venv/bin/activate验证:
python -c "import torch; print(torch.__version__); print(torch.version.cuda); print(torch.cuda.is_available())"
python -c "import deepxde as dde; print(dde.__version__)"如果已经有虚拟环境:
python -m pip install -r requirements.txt如果从零开始:
cd /home/guiyu/workspace/PINN
python3 -m venv .venv
source .venv/bin/activate
python -m pip install --upgrade pip
python -m pip install -r requirements.txt说明:
普通 Python 包使用清华 PyPI 镜像。
torch 使用 PyTorch 官方 CPU wheel 源,避免下载 NVIDIA CUDA/cuDNN 依赖。
手写 PyTorch 版本:
cd /home/guiyu/workspace/PINN
source .venv/bin/activate
python pinn_ode_intro_torch.pyDeepXDE 版本:
python pinn_intro_deepXDE.pyParker 太阳风 DeepXDE 快速验证:
python pinn_parker_solar_wind_deepXDE.py更完整的运行检查流程见:
TESTING.md
该脚本求解:
du/dt = -u
u(0) = 1
解析解:
u(t) = exp(-t)
运行后会输出训练损失和误差,并保存图像到:
outputs/pinn_ode_intro_result.png
当前已跑通结果:
Relative L2 error: 1.293063e-03
Max absolute error: 1.406372e-03
PINN_入门教程.ipynb Jupyter Notebook 入门教程
pinn_ode_intro_torch.py PyTorch 手写 PINN ODE 示例
pinn_intro_deepXDE.py DeepXDE PINN ODE 示例
pinn_parker_solar_wind_deepXDE.py DeepXDE Parker 太阳风快速验证
PINN_DeepXDE_速查手册.md 机器学习、PINN、DeepXDE 速查文档
TESTING.md 运行与测试方法
requirements.txt pip 依赖文件
outputs/ 运行结果图像
建议按这个顺序扩展:
1. 一阶 ODE:du/dt = -u
2. 二阶 ODE:u'' + u = 0
3. 一维热传导方程:u_t = alpha u_xx
4. Burgers 方程:u_t + u u_x = nu u_xx
5. Parker 太阳风方程
6. MHD 表面输运模型