Skip to content

Deep Learning

Cori GPU provides several modules to different machine learning and deep learning tools and libraries.

cuDNN

cuDNN is NVIDIA's library of GPU-accelerated primitives for deep neural networks. It is available as the cudnn module, and is loaded automatically when a cuda module is loaded.

TensorFlow

TensorFlow is an open-source toolkit for machine learning. It is available as the module tensorflow/gpu-<version>. Users are also encouraged to use GPU-optimized TensorFlow containers provided on NVIDIA's NGC container library. NGC containers can run on Cori GPU using Shifter.

PyTorch

PyTorch is an open-source machine learning framework. It is available as the module pytorch/<version>-gpu. Users are also encouraged to use GPU-optimized PyTorch containers from NGC.