Caffe2 Release 17.04
The NVIDIA container image of Caffe2, release 17.04, is available.
Caffe2 container image version 17.04 is based on Caffe2 0.5.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
- NVIDIA CUDA® Deep Neural Network library™ (cuDNN) 6.0.20
- NCCL 1.6.1 (optimized for NVLink™ )
- OpenMPI™ 1.10.3
Key Features and Enhancements
This Caffe2 release includes the following key features and enhancements.
- Examples that support multi-GPU using NCCL for computer vision, such as, image classification, segmentation, and more:
- MNIST, CIFAR-10, and IMAGENET datasets
Note:
Note: IMAGENET is the dataset on which neural networks, such as, AlexNet, VGG, OverFeat, ResNet, and Inception are trained on. These neural networks are comprised of different depths and complexities.
/workspace/nvidia-examples
directory.
- MNIST, CIFAR-10, and IMAGENET datasets
- Ubuntu 16.04 with March 2017 updates
Known Issues
There are no known issues in this release.