The NGC containers v18.01 for popular deep learning frameworks are now available at https://ngc.nvidia.com
These containers have support for:
- Python 2 and/or Python 3
- Latest version of cuBLAS and cuDNN
- Caffe2 0.8.1
- NVCaffe 0.16.5
- DIGITS 6.0
- Microsoft Cognitive Toolkit 2.3.1
- MXNet 1.0
- PyTorch 0.3
- TensorFlow 1.4
- TensorRT 3.0.1
- Theano 1.0.1
- Torch7
- Ubuntu 16.04 with December 2017 updates
For the current release notes, see http://docs.nvidia.com/deeplearning/dgx/
To view the supported deep learning frameworks matrix, see: Frameworks Support Matrix :: NVIDIA Deep Learning Frameworks Documentation