Running a Python script on LANTA HPC using Mamba

Owned by Yutthana Wongnongwa
Last updated: Mar 29, 2024
2 min read

This article describes how to execute a Python script on the LANTA HPC system using Miniconda. The following table of contents provides an overview of the article's material, allowing for quick identification of the most relevant sections.

Slurm script example for running the Python script

Running on Compute node

#!/bin/bash #SBATCH -p compute # Specify partition [Compute/Memory/GPU] #SBATCH -N 1 -c 128 # Specify number of nodes and processors per task #SBATCH --ntasks-per-node=1 # Specify tasks per node #SBATCH -t 120:00:00 # Specify maximum time limit (hour: minute: second) #SBATCH -A ltxxxxxx # Specify project name #SBATCH -J JOBNAME # Specify job name module load Mamba/23.11.0-0 # Load the conda module conda activate tensorflow-2.12.1 # Activate your environment python3 file.py # Run your program or executable code

Full node: -c 128, Half node: -c 64, ¼ node: -c 32

Running on GPU node

#!/bin/bash #SBATCH -p gpu # Specify partition [Compute/Memory/GPU] #SBATCH -N 1 -c 16 # Specify number of nodes and processors per task #SBATCH --gpus-per-task=1 # Specify number of GPU per task #SBATCH --ntasks-per-node=4 # Specify tasks per node #SBATCH -t 120:00:00 # Specify maximum time limit (hour: minute: second) #SBATCH -A ltxxxxxx # Specify project name #SBATCH -J JOBNAME # Specify job name module load Mamba/23.11.0-0 # Load the conda module conda activate tensorflow-2.12.1 # Activate your environment python3 file.py # Run your program or executable code

1 GPU card: --ntasks-per-node=1, 2 GPU cards: --ntasks-per-node=2, 4 GPU cards: --ntasks-per-node=4

Submit a job

Use the sbatch script.sh command to submit your job to the Slurm system.

username@lanta:~> sbatch script.sh

Before you use the sbatch script.sh command, you must ensure that your environment is disabled.

Related articles