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 126.96.36.199 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
imagentscript in the
- Ubuntu 16.04 with June 2017 updates
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.