Caffe2 Release 17.07
The NVIDIA container image of Caffe2, release 17.07, is available.
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
- 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.