Caffe2 Release 17.07

The NVIDIA container image of Caffe2, release 17.07, is available.

Caffe2 container image version 17.07 is based on Caffe2 0.7.0.

Contents of Caffe2

This container image contains the complete source of the version of Caffe2 in /opt/caffe2. It is pre-built and installed into the /opt/caffe2/[binaries,lib] directories in the container image. The container also includes the following:

• Ubuntu 16.04
• NVIDIA CUDA 8.0.61.2 including CUDA^® Basic Linear Algebra Subroutines library™ (cuBLAS) Patch 2
• NVIDIA CUDA^® Deep Neural Network library™ (cuDNN) 6.0.21
• NCCL 2.0.3 (optimized for NVLink™ )
• OpenMPI™ 1.10.3

Key Features and Enhancements

This Caffe2 release includes the following key features and enhancements.

• Updated version of the imagent script in the nvidia-examples directory.
• Ubuntu 16.04 with June 2017 updates

Known Issues

You may come across unexpected behavior, including a runtime cash, when using the predictor_exporter.py script to save or load models or checkpoints. However, in this release, there may be cases where the script can save or load a model incorrectly. The imagent script at nvidia-examples/imagenet/train.py uses the predictor_exporter.py script, therefore, when using the imagent script, you may come across unexpected behavior. Alternatively, use the SaveOp and LoadOp operations, with the correct argument, to save and load the models and checkpoints.