The conda environment with the TF package and other Python dependencies need to be installed using the following install-commands first.


module load anaconda
conda create -n lhapdf-tf python=3.8 pandas matplotlib
source activate lhapdf-tf
pip install tensorflow-gpu==2.8 

Then make sure that the .slurm or .sh file contains the following content with the correct file name.

#!/usr/bin/env bash
#SBATCH -p standard
#SBATCH --output=result_%a.out
#SBATCH -c 1
#SBATCH -t 16:30:00
#SBATCH -A spinquest

module load anaconda cuda/11.4.2 cudnn/8.2.4.15
module load gcc/9.2.0 lhapdf/6.3.0
source activate lhapdf-tf
export PYTHONPATH=~/.conda/envs/lhapdf-tf/lib/python3.8/site-packages:$PYTHONPATH
python <File_Name.py>

  • No labels