This example demonstrates how to set up a data-driven model for a 2D Darcy flow using the Fourier Neural Operator (FNO) architecture inside of PhysicsNeMo. This example runs on a single GPU, go to the darcy_nested_fno example for exploring a multi-GPU training.

Install the required dependencies by running below:

pip install -r requirements.txt

To train the model, run

python train_fno_darcy.py

training data will be generated on the fly.

• Fourier Neural Operator for Parametric Partial Differential Equations