The NGC containers v18.05 for popular deep learning frameworks are now available at https://ngc.nvidia.com
These containers have support for:
- Python 2 and/or Python 3
- CUDA 9
- cuBLAS v9.0.333
- cuDNN 7.1.2
- NCCL 2.1.15
- Caffe2 0.8.1
- NVCaffe 0.17.0
- DIGITS 6.1.1
- Microsoft Cognitive Toolkit 2.5
- Inference Server 1.0.0
- MXNet 1.1.0
- PyTorch 0.4.0
- TensorFlow 1.7.0
- TensorRT 3.0.4
- Theano 1.0.1
- Torch7
- Ubuntu 16.04 with April 2018 updates
For the current release notes, see http://docs.nvidia.com/deeplearning/dgx/
To view the support matrix across all optimized containers, see: https://docs.nvidia.com/deeplearning/dgx/support-matrix/index.html.