...
1. module purge
2. module load Python cuDNN
3. cd $PROJ
4. python -m venv myTensorflow
5. source myTensorflow/bin/activate
6. pip --cache-dir </wrk-[vakka|kappa]/users/<username>> install tensorflow
7. pip --cache-dir </wrk-[vakka|kappa]/users/<username>> install tensorflow-gpu
You can then add additional libraries etc..
8. pip --cache-dir </wrk-[vakka|kappa]/users/<username>> install keras
9. pip --cache-dir </wrk-[vakka|kappa]/users/<username>> install <any additional packages you wish to install...>
...
#!/bin/bash
############## This section states the requirements the job requires:
#SBATCH --job-name=test
#SBATCH --workdir=$WRKDIR[vakka|kappa]
#SBATCH -o result.txt
#SBATCH -p gpu
#SBATCH -c 2
#SBATCH --gres=gpu:1
#SBATCH --mem-per-cpu=100
############## Here starts the actual UNIX commands and payload:
module purge
module load Python cuDNN
source myTensorflow/bin/activate
srun tensorflow-program
...