The NGC containers v18.02 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.282; including cuBLAS 9.0.234 Patch 1
- Caffe2 0.8.1
- NVCaffe 0.16.5
- DIGITS 6.1
- Microsoft Cognitive Toolkit 2.3.1
- MXNet 1.0.0
- PyTorch 0.3.0
- TensorFlow 1.4.0
- TensorRT 3.0.4
- Theano 1.0.1
- Torch7
- Ubuntu 16.04 with January 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