bug fix on 2026-03-30

Author: nle18370

README.md

@@ -26,6 +26,36 @@ Created and maintained by Makoto Takamoto
 Francesco Alesiani, Dirk Pflüger, and Mathias Niepert.
 
 ---
+## Compatibility with Newer Python, JAX, PyTorch, and CUDA Versions
+
+While this project officially targets older versions: 
+- Python 3.9  
+- JAX 0.4.11  
+- PyTorch 1.13.0  
+- CUDA 11.7  
+
+we have verified that most components remain compatible with newer releases.:
+- Python 3.12  
+- JAX 0.9.2  
+- PyTorch 2.11.0  
+- CUDA 13.0  
+
+In particular, the core codebase, forward training in `pdebench/model`, and data generation in `data_gen_NLE` work with these newer versions.
+Compatibility for the remaining components is still under investigation.
+_Last updated: 2026-03-30_
+
+## Data Path Issue in Forward Training (`model/fno/utils.py` and `model/unet/utils.py`)
+
+We identified an issue with data paths during forward training in `model/fno/utils.py` and `model/unet/utils.py`. This problem is caused by Hydra modifying the `data_path` at runtime.
+
+To resolve this issue, please use the commented lines:
+- Lines 185–188 in `model/fno/utils.py`  
+- Lines 185–187 in `model/unet/utils.py`  
+
+(Note that the above modification (use of "to_absolute_path" in hydra) is allowed Hydra version >= 0.11.0.)
+
+_Last updated: 2026-03-30_
+
 
 ## Datasets and Pretrained Models
 

pdebench/data_gen/data_gen_NLE/CompressibleFluid/CFD_multi_Hydra.py

@@ -159,6 +159,8 @@
 import jax.numpy as jnp
 from jax import device_put, jit, lax
 from omegaconf import DictConfig
+
+sys.path.append("..")  # if PDEBench cannot recognize the location of utils
 from utils import (
     Courant_HD,
     Courant_vis_HD,
@@ -180,8 +182,6 @@
 os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
 os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"] = ".9"
 
-sys.path.append("..")
-
 # if double precision
 # from jax.config import config
 # config.update("jax_enable_x64", True)

pdebench/data_gen/data_gen_NLE/CompressibleFluid/run_trainset_1D.sh

@@ -3,8 +3,8 @@ nn=1
 key=2020
 while [ "$nn" -le 10 ]; do
   CUDA_VISIBLE_DEVICES='0,1' python3 CFD_multi_Hydra.py +args=1D_Multi.yaml ++args.init_key="$key"
-  nn=$(${nn} + 1)
-  key=$(${key} + 1)
+  nn=$((nn + 1))
+  key=$((key + 1))
   echo "$nn"
   echo "$key"
 done

pdebench/data_gen/data_gen_NLE/CompressibleFluid/run_trainset_1DShock.sh

@@ -3,8 +3,8 @@ nn=1
 key=2031
 while [ "$nn" -le 10 ]; do
   CUDA_VISIBLE_DEVICES='0,1' python3 CFD_multi_Hydra.py +args=1D_Multi_shock.yaml ++args.init_key="$key"
-  nn=$(${nn} + 1)
-  key=$(${key} + 1)
+  nn=$((nn + 1))
+  key=$((key + 1))
   echo "$nn"
   echo "$key"
 done

pdebench/data_gen/data_gen_NLE/CompressibleFluid/run_trainset_1D_trans.sh

@@ -3,8 +3,8 @@ nn=1
 key=2020
 while [ "$nn" -le 10 ]; do
   CUDA_VISIBLE_DEVICES='0,1' python3 CFD_multi_Hydra.py +args=1D_Multi_trans.yaml ++args.init_key="$key"
-  nn=$(${nn} + 1)
-  key=$(${key} + 1)
+  nn=$((nn + 1))
+  key=$((key + 1))
   echo "$nn"
   echo "$key"
 done

pdebench/data_gen/data_gen_NLE/CompressibleFluid/run_trainset_2D.sh

@@ -4,8 +4,8 @@ key=2031
 while [ "$nn" -le 100 ]; do
   CUDA_VISIBLE_DEVICES='0,1,2,3' python3 CFD_multi_Hydra.py +args=2D_Multi_Rand.yaml ++args.init_key="$key"
   #CUDA_VISIBLE_DEVICES='0,1,2,3' python3 CFD_multi_Hydra.py +args=2D_Multi_Rand_HR.yaml ++args.init_key="$key"
-  nn=$(${nn} + 1)
-  key=$(${key} + 1)
+  nn=$((nn + 1))
+  key=$((key + 1))
   echo "$nn"
   echo "$key"
 done

pdebench/data_gen/data_gen_NLE/CompressibleFluid/run_trainset_2DTurb.sh

@@ -4,8 +4,8 @@ key=2031
 #while [ "$nn" -le 100 ]; do
 while [ "$nn" -le 55 ]; do
   CUDA_VISIBLE_DEVICES='0,1,2,3' python3 CFD_multi_Hydra.py +args=2D_Multi_Turb.yaml ++args.init_key="$key"
-  nn=$(${nn} + 1)
-  key=$(${key} + 1)
+  nn=$((nn + 1))
+  key=$((key + 1))
   echo "$nn"
   echo "$key"
 done

pdebench/data_gen/data_gen_NLE/CompressibleFluid/run_trainset_3D.sh

@@ -3,8 +3,8 @@ nn=1
 key=2031
 while [ "$nn" -le 10 ]; do
   CUDA_VISIBLE_DEVICES='0,1' python3 CFD_multi_Hydra.py +args=3D_Multi_Rand.yaml ++args.init_key="$key"
-  nn=$(${nn} + 1)
-  key=$(${key} + 1)
+  nn=$((nn + 1))
+  key=$((key + 1))
   echo "$nn"
   echo "$key"
 done

pdebench/data_gen/data_gen_NLE/CompressibleFluid/run_trainset_3DTurb.sh

@@ -3,8 +3,8 @@ nn=1
 key=2031
 while [ "$nn" -le 6 ]; do
   CUDA_VISIBLE_DEVICES='0,1,2,3' python3 CFD_multi_Hydra.py +args=3D_Multi_TurbM1.yaml ++args.init_key="$key"
-  nn=$(${nn} + 1)
-  key=$(${key} + 1)
+  nn=$((nn + 1))
+  key=$((key + 1))
   echo "$nn"
   echo "$key"
 done

pdebench/data_gen/data_gen_NLE/ReactionDiffusionEq/run_DarcyFlow2D.sh

@@ -4,8 +4,8 @@ key=2020
 while [ "$nn" -le 50 ]; do
   CUDA_VISIBLE_DEVICES='0,1' python3 reaction_diffusion_2D_multi_solution_Hydra.py +multi=config_2D.yaml ++multi.init_k\
 ey="$key"
-  nn=$(${nn} + 1)
-  key=$(${key} + 1)
+  nn=$((nn + 1))
+  key=$((key + 1))
   echo "$nn"
   echo "$key"
 done