-
Notifications
You must be signed in to change notification settings - Fork 54
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
removed some links to data, added slurm script examples and readme
- Loading branch information
Hoppe
committed
Aug 29, 2024
1 parent
97de398
commit fef1477
Showing
4 changed files
with
28 additions
and
9 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,12 @@ | ||
There are two example scripts in this directory, `slurm_script_cpu.sh` and `slurm_script_gpu.sh`, that demonstrate how to run a Heat application on an HPC-system with SLURM as resource manager. | ||
|
||
1. `slurm_script_cpu.sh` is an example script that runs a Heat application on a CPU node. You must specify the name of the respective partition of your cluster. Moreover, the | ||
numer of CPU cores available at a node of your system must be greater or equal to the product of the tasks-per-node- and the cpus-per-task-argument (=8x16=128 in the example). | ||
|
||
2. `slurm_script_gpu.sh` is an example script that runs a Heat application on a GPU node. You must specify the name of the respective partition of your cluster. Moreover, the | ||
numer of GPU devices available at a node of your system must be greater or equal to the number of GPUs requested in the script (=4 in the example). | ||
|
||
## Remarks | ||
|
||
* Please have a look into the documentation of your HPC-system for its detailed configuration and properties. Maybe, you have to adjust the script to your system. | ||
* You need to load the required modules (e.g., for MPI, CUDA etc.) from the module system of your HPC-system before running the script. Moreover, you need to install Heat in a virtual environment (and activate it). Alternatively, you may use spack (if available on your system) for installing Heat and its dependencies. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,12 @@ | ||
#!/bin/bash | ||
|
||
#SBATCH --partition=<partition_name> | ||
#SBATCH --nodes=1 | ||
#SBATCH --tasks-per-node=8 | ||
#SBATCH --cpus-per-task=16 | ||
#SBATCH --time="00:01:00" | ||
|
||
export MKL_NUM_THREADS=$SLURM_CPUS_PER_TASK | ||
export OMP_NUM_THREADS=$SLURM_CPUS_PER_TASK | ||
|
||
srun python ~/heat/tutorials/hpc/01_basics/01_basics_dndarrays.py |
5 changes: 2 additions & 3 deletions
5
tutorials/hpc/run_script.sh → tutorials/hpc/slurm_script_gpu.sh
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters