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.




User Information

Connect to PSC systems:
Technical questions:

Send mail to or call the PSC hotline: 412-268-6350.