Abstract

This support matrix is for NVIDIA optimized frameworks. The matrix provides a single view into the supported software and specific versions that come packaged with the frameworks based on the container image.

1. Support Matrix For NVIDIA Optimized Frameworks

Important: Content that is included in <<>> brackets indicates new content from the previously published version.
Note: The deep learning framework container packages follow a naming convention that is based on the year and month of the image release. For example, the 18.01 release of an image was released in January, 2018.

18.xx container images

Table 1. Software stack packaged with the 18.xx container images
  Container Image 18.04 18.03 18.02 18.01
Supported Platform DGX OS 3.1.2+ and 2.1.1+ 3.1.2+ and 2.1.1+ 3.1.2+ and 2.1.1+ 3.1.2+ and 2.1.1+
NVIDIA Driver 384 384 384 384
Base Image Ubuntu 16.04 16.04 16.04 16.04
CUDA 9.0.176 9.0.176 9.0.176 9.0.176
cuBLAS 9.0.333 9.0.333 9.0.282 Patch 2 and cuBLAS 9.0.234 Patch 1 9.0.282 Patch 2
cuDNN 7.1.1 7.1.1 7.0.5 7.0.5
NCCL <<2.1.15>> 2.1.2 2.1.2 2.1.2
NVIDIA Optimized Frameworks NVCaffe <<0.17.0>> and Python 2.7 0.16.6 and Python 2.7 0.16.5 and Python 2.7 0.16.5 and Python 2.7
Caffe2 0.8.1 including 0.8.1 including 0.8.1 including 0.8.1 including
DIGITS <<6.1.1>> 6.1.0 6.1.0 6.0.0
Microsoft Cognitive Toolkit 2.4 including 2.4 including 2.3.1 including 2.3.1 including
<<Inference Server>> <<0.0.1 Beta Release>>      
MXNet 1.1.0and Python 2.7 or Python 3.5 1.1.0and Python 2.7 or Python 3.5 1.0.0 and Python 2.7 or Python 3.5 1.0.0 and Python 2.7 or Python 3.5
PyTorch <<0.3.1>> and Python 3.6 0.3.0 and Python 3.6 0.3.0 and Python 3.6 0.3.0 and Python 3.6
TensorFlow <<1.7.0>> including 1.4.0 including 1.4.0 including 1.4.0 including
TensorRT 3.0.4 and Python 2.7 3.0.4 and Python 2.7 3.0.4 and Python 2.7 3.0.1 and Python 2.7
Theano 1.0.1 and Python 2.7 1.0.1 and Python 2.7 1.0.1 and Python 2.7 1.0.1 and Python 2.7
Torch 7 and Python 2.7 7 and Python 2.7 7 and Python 2.7 7 and Python 2.7

17.xx container images

Table 2. Software stack packaged with the 17.xx container images
  Container Image 17.12 17.11 17.10 17.09 17.07 17.06 17.05 17.04 17.03 17.02 17.01
Supported Platform DGX OS 3.1.2+ and 2.1.1+ 3.1.2+ and 2.1.1+ 3.1.2+ and 2.1.1+ 3.1.2+ and 2.1.1+ 2.x+ and 1.x+ 2.x+ and 1.x+ 2.x+ and 1.x+ 2.x+ and 1.x+ 2.x+ and 1.x+ 2.x+ and 1.x+ 2.x+ and 1.x+
NVIDIA Driver 384 384 384 384              
Base Image Ubuntu 16.04 16.04 16.04 16.04 16.04 16.04 16.04 16.04 16.04 14.04 14.04
CUDA 9.0.176 9.0.176 9.0 9.0 8.0.61.2 8.0.61 8.0.61 8.0.61 8.0.61 8.0.61 8.0.54
cuBLAS 9.0.234 9.0.234     patch 2            
cuDNN 7.0.5 7.0.4 7.0.3 7.0.2 6.0.21 6.0.21 6.0.21 6.0.20 6.0.20 6.0.13 6.0.10
NCCL 2.1.2 2.1.2 2.0.5 2.0.5 2.0.3 1.6.1 1.6.1 1.6.1 1.6.1 1.6.1 1.6.1
NVIDIA Optimized Frameworks NVCaffe 0.16.4 0.16.4 0.16.4 0.16.4 0.16 0.16 0.16 0.16 0.16 0.16 0.16
Caffe2 0.8.1 and OpenMPI 1.10.3 0.8.1 and OpenMPI 1.10.3 0.8.1 and OpenMPI 1.10.3 0.8.1 and OpenMPI 1.10.3 0.7.0 and OpenMPI 1.10.3 0.7.0 and OpenMPI 1.10.3 0.5.0+ and OpenMPI 1.10.3 0.5.0+ and OpenMPI 1.10.3      
DIGITS 6.0.0 6.0.0 6.0.0 6.0.0 6.0.0 5.0 5.0 5.0 5.0 5.0 5.0
Microsoft Cognitive Toolkit 2.2 and OpenMPI 3.0.0 2.2 and OpenMPI 3.0.0 2.2 2.1 2.0 2.0 2.0.rc2 2.0.beta15.0 2.0.beta12.0 2.0.beta9.0 2.0.beta5.0
MXNet 1.0.0 0.12.0 0.11.0 0.11.0.rc3 0.10.0 0.10.0 0.9.3a+ 0.9.3a+ 0.9.3    
PyTorch 0.2.0 0.2.0 0.2.0 0.2.0 0.1.12 0.1.12 0.1.12 0.1.10      
TensorFlow 1.4.0 1.3.0 1.3.0 1.3.0 1.2.1 1.1.0 1.0.1 1.0.1 1.0.0 0.12.1 0.12.0
TensorRT 3.0.1                    
Theano 1.0.0rc1 1.0.0rc1 0.10beta3 0.10beta1 0.9.0 0.9.0 0.9.0 0.9.0 0.9.0rc3 0.8.0 0.8.0
Torch 7 7 7 7 7 7 7 7 7 7 7

16.xx container images

Table 3. Software stack packaged with the 16.xx container images
  Container Image 16.12
Supported Platform DGX OS 2.x+ and 1.x+
NVIDIA Driver  
Base Image Ubuntu 14.04
CUDA 8.0.54
cuBLAS  
cuDNN 6.0.5
NCCL 1.6.1
NVIDIA Optimized Frameworks NVCaffe 0.16
Caffe2  
DIGITS 5.0
Microsoft Cognitive Toolkit 2.0.beta5.0
MXNet  
PyTorch  
TensorFlow 0.12.0
TensorRT  
Theano 0.8.0
Torch 7

Notices

Notice

THE INFORMATION IN THIS GUIDE AND ALL OTHER INFORMATION CONTAINED IN NVIDIA DOCUMENTATION REFERENCED IN THIS GUIDE IS PROVIDED “AS IS.” NVIDIA MAKES NO WARRANTIES, EXPRESSED, IMPLIED, STATUTORY, OR OTHERWISE WITH RESPECT TO THE INFORMATION FOR THE PRODUCT, AND EXPRESSLY DISCLAIMS ALL IMPLIED WARRANTIES OF NONINFRINGEMENT, MERCHANTABILITY, AND FITNESS FOR A PARTICULAR PURPOSE. Notwithstanding any damages that customer might incur for any reason whatsoever, NVIDIA’s aggregate and cumulative liability towards customer for the product described in this guide shall be limited in accordance with the NVIDIA terms and conditions of sale for the product.

THE NVIDIA PRODUCT DESCRIBED IN THIS GUIDE IS NOT FAULT TOLERANT AND IS NOT DESIGNED, MANUFACTURED OR INTENDED FOR USE IN CONNECTION WITH THE DESIGN, CONSTRUCTION, MAINTENANCE, AND/OR OPERATION OF ANY SYSTEM WHERE THE USE OR A FAILURE OF SUCH SYSTEM COULD RESULT IN A SITUATION THAT THREATENS THE SAFETY OF HUMAN LIFE OR SEVERE PHYSICAL HARM OR PROPERTY DAMAGE (INCLUDING, FOR EXAMPLE, USE IN CONNECTION WITH ANY NUCLEAR, AVIONICS, LIFE SUPPORT OR OTHER LIFE CRITICAL APPLICATION). NVIDIA EXPRESSLY DISCLAIMS ANY EXPRESS OR IMPLIED WARRANTY OF FITNESS FOR SUCH HIGH RISK USES. NVIDIA SHALL NOT BE LIABLE TO CUSTOMER OR ANY THIRD PARTY, IN WHOLE OR IN PART, FOR ANY CLAIMS OR DAMAGES ARISING FROM SUCH HIGH RISK USES.

NVIDIA makes no representation or warranty that the product described in this guide will be suitable for any specified use without further testing or modification. Testing of all parameters of each product is not necessarily performed by NVIDIA. It is customer’s sole responsibility to ensure the product is suitable and fit for the application planned by customer and to do the necessary testing for the application in order to avoid a default of the application or the product. Weaknesses in customer’s product designs may affect the quality and reliability of the NVIDIA product and may result in additional or different conditions and/or requirements beyond those contained in this guide. NVIDIA does not accept any liability related to any default, damage, costs or problem which may be based on or attributable to: (i) the use of the NVIDIA product in any manner that is contrary to this guide, or (ii) customer product designs.

Other than the right for customer to use the information in this guide with the product, no other license, either expressed or implied, is hereby granted by NVIDIA under this guide. Reproduction of information in this guide is permissible only if reproduction is approved by NVIDIA in writing, is reproduced without alteration, and is accompanied by all associated conditions, limitations, and notices.

Trademarks

NVIDIA, the NVIDIA logo, and cuBLAS, CUDA, cuDNN, cuFFT, cuSPARSE, DIGITS, DGX, DGX-1, DGX Station, GRID, Jetson, Kepler, NVIDIA GPU Cloud, Maxwell, NCCL, NVLink, Pascal, Tegra, TensorRT, Tesla and Volta are trademarks and/or registered trademarks of NVIDIA Corporation in the Unites States and other countries. Other company and product names may be trademarks of the respective companies with which they are associated.