Documentation access will be interrupted from time to time due to some bug correction.

Page tree
Skip to end of metadata
Go to start of metadata

This page describes how to prepare for and run jobs which use Tensorflow on the GPU nodes.

First you need to clone the following repository:

git clone

The currently available version is 1.9.0. New version will be added once they are available.The recommended setup uses Python virtual environments, as they enable Tensorflow to be installed in an isolated and stable environment.

One time setup

The only supported Python at the moment is the system one, Python 3.6.5.

These commands have to be executed within a job in the gpu partition and QOS (alternatively it's also possible to load the GPU software environment in any other node).

Sinteract -p gpu -q gpu -t 01:00:00 -m 4G
module purge
module load gcc cuda cudnn python 
virtualenv --system-site-packages -p python3 venv-tensorflow-1.9
source venv-tensorflow-1.9/bin/activate
pip3 install --upgrade pip setuptools
pip3 install --no-cache-dir --upgrade tensorflow-pip-deneb/1.9-gpu/tensorflow-1.9.0-cp36-cp36m-linux_x86_64.whl

Additionally, if other Python packages are needed they should be installed inside the same Python virtual environment. Python packages installed in the system or the user home directory are not available within the Python virtual environment.

For example, if pandas is needed, use the following command (still within a job in the gpu partition and with at least the same modules loaded):

pip3 install --no-cache-dir --upgrade pandas

Running a job in batch mode

Sbatch your_job_script

Example job script:

#!/bin/bash -l
#SBATCH --nodes=1
#SBATCH --time=1:0:0
#SBATCH --qos=gpu
#SBATCH --gres=gpu:1
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=1
#SBATCH --partition=gpu

slmodules -s x86_E5v2_Mellanox_GPU
module load gcc cuda cudnn mvapich2 openblas
source venv-tensorflow-1.9/bin/activate
srun python

Running a job interactively

iPython is needed. It should be installed inside the same Python virtual environment as above using the following command:

pip3 install --no-cache-dir --upgrade ipython


Sinteract -p gpu -q gpu_free -g gpu:1

module purge
module load gcc cuda cudnn python

source venv-tensorflow-1.9/bin/activate


copy/paste the lines from