Caffe is a deep learning framework made with expression, speed, and modularity in mind. It is developed by Berkeley AI Research (BAIR) and by community contributors.

Caffe is available within Singularity container

Please check how to build caffe (CPU only) container with singularity

We will use caffe-gpu.img container image.

Here is an example of SGE script that use caffe GPU version withing singularity:

#!/bin/bash 
 
#$ -o $JOB_NAME.o$JOB_ID
 
#$ -N job_caffe
 
i#$ -q tesla.q   ## we request tesla.q             
 
#$ -l h_vmem=8G   ## we request 8G
 
 
## load singularity
 
module load tools/singularity
 
## run TF with singularity
 
singularity exec --nv $SING_SPOOL/caffe-gpu.img caffe train -solver /opt/caffe/examples/mnist/lenet_solver.prototxt

Open a terminal and connect to GPU node with qlogin:

$ qlogin -q tesla.q -l h_vmem=8G
node1-50:~$ module load tools/singularity
node1-50:~$ singularity shell --nv $SING_SPOOL/caffe-gpu.img
singularity:~> caffe train -solver /opt/caffe/examples/mnist/lenet_solver.prototxt