You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Once ClimaAtmos#2341 is finalized and merged, we can add full GPU support.
This will just require setting proper sbatch arguments for the driver, pipeline.sh. We could do this quickly by adding an additional GPU driver, but I would rather have a unified interface.
I am planning on doing this by introducing a proper command line interface, which will also allow a simpler interface for running experiments.
We will need to :
determine the correct GPU settings. This will be quick.
Add an argument parser and change the way the pipeline takes in configurations. This will be the bulk of the work, but won't be too complicated because the pipeline is still quite simple.
The text was updated successfully, but these errors were encountered:
nefrathenrici
changed the title
GPU Support + CLI Options
Allow per-experiment slurm configuration
Mar 14, 2024
Once ClimaAtmos#2341 is finalized and merged, we can add full GPU support.
This will just require setting proper
sbatch
arguments for the driver,pipeline.sh
. We could do this quickly by adding an additional GPU driver, but I would rather have a unified interface.I am planning on doing this by introducing a proper command line interface, which will also allow a simpler interface for running experiments.
We will need to :
The text was updated successfully, but these errors were encountered: