- Login to vsc3
- Load Anaconda:
module load anaconda3/5.3.0
- Install TensorFlow GPU version:
conda create -n tf-gpu tensorflow-gpu=1.15
- Activate:
conda activate tf-gpu
- Install the required modules
- Deactivate:
conda deactivate
- Create a shell script:
train_dhSegment.sh
(Shell script is needed, because training cannot be started directly since the login server has no GPU support)
#!/bin/bash
#SBATCH -J tensorflow_test_fabian
#SBATCH -N 1
#SBATCH --partition gpu_gtx1080single
#SBATCH --qos gpu_gtx1080single
#SBATCH --gres gpu:1
module purge
module load anaconda3/5.3.0
module load cuda/10.0.130
source /opt/sw/x86_64/glibc-2.17/ivybridge-ep/anaconda3/5.3.0/etc/profile.d/conda.sh
conda activate tf-gpu
cd dhSegment
python3 train.py with demo/demo_config_2.json
- Submit the job:
sbatch train_dhSegment.sh
- Monitor the job with
squeue --job [jobid]