Caffe2 is a lightweight, modular and scalable deep learning framework.
Singularity containers for Caffe2 are available for use on Bridges in directory /pylon5/containers/ngc/caffe2. Multiple containers are available, for different versions of python and supporting software. These containers can be used on the Volta or P100 GPU nodes. See Singularity images on Bridges for more information on the available containers.
To see what versions of Caffe2 are available type
module avail caffe2
To see what other modules are needed, what commands are available and how to get additional help type
module help caffe2
To use Caffe, include a command like this in your batch script or interactive session to load the caffe module:
module load caffe2
Be sure you also load any other modules needed, as listed by the
module help caffe2 command.