The NGC containers v18.03 for popular deep learning frameworks are now available at https://ngc.nvidia.com
These containers have support for:
- Python 2 and/or Python 3
- cuBLAS v9.0.333
- cuDNN 7.1.1
- Caffe2 0.8.1
- NVCaffe 0.16.6
- DIGITS 6.1
- Microsoft Cognitive Toolkit 2.4
- MXNet 1.1.0
- PyTorch 0.3.0
- TensorFlow 1.4.0
- TensorRT 3.0.4
- Theano 1.0.1
- Torch7
- Ubuntu 16.04 with February 2018 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