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>