diff --git a/docs/source/resources/support_matrix.rst b/docs/source/resources/support_matrix.rst index 18a7c94f6f5..190eb957333 100644 --- a/docs/source/resources/support_matrix.rst +++ b/docs/source/resources/support_matrix.rst @@ -19,60 +19,60 @@ The following tables provide the software and model versions that NVTabular supp :raw-html:`

Table 1: Support matrix for the Merlin Inference (merlin-inference) container

` -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| | **Version 0.6** | **Version 21.09** | **Version 21.11** | **Version 21.12** | **Version 22.01** | **Version 22.02** | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| **DGX** | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| DGX System | * DGX-1 | -| | * DGX-2 | -| | * DGX A100 | -| | * DGX Station | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| Operating System | Ubuntu version 20.04 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| **NVIDIA Certified Systems** | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| NVIDIA Driver | NVIDIA Driver version 465.19.01 or later is required. However, if you're | -| | running on Data Center GPUs (formerly Tesla) such as T4, you can use | -| | any of the following NVIDIA Driver versions: | -| | | -| | * 418.40 (or later R418) | -| | * 440.33 (or later R440) | -| | * 450.51 (or later R450) | -| | * 460.27 (or later R460) | -| | | -| | **NOTE**: The CUDA Driver Compatibility Package doesn’t support all | -| | drivers. | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| GPU Model | * `NVIDIA Ampere GPU Architecture | -| | `_ | -| | * `Turning `_ | -| | * `Volta | -| | `_ | -| | * `Pascal | -| | `_ | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| **Base Container Image** | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| Container Operating System | Ubuntu version 20.04 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| Base Container | Triton version 21.07 | Triton version 21.10 | Triton version 21.11 | Triton version 21.12 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| CUDA | 11.6 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| RMM | 21.06 | 21.08 | 21.10 | 21.12 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| cuDF | 21.06 | 21.08 | 21.10 | 21.12 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| cuDNN | N/A | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| HugeCTR | 3.1 | 3.2 | 3.2.1 | 3.3 | 3.3.1 | 3.4 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| NVTabular | 0.6 | 0.7 | 0.7.1 | 0.8 | 0.9 | 0.10 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| SM | 60; 61; 70; 75; 80 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| | **Version 21.09** | **Version 21.11** | **Version 21.12** | **Version 22.01** | **Version 22.02** | **Version 22.03** | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| **DGX** | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| DGX System | * DGX-1 | +| | * DGX-2 | +| | * DGX A100 | +| | * DGX Station | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| Operating System | Ubuntu version 20.04 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| **NVIDIA Certified Systems** | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| NVIDIA Driver | NVIDIA Driver version 465.19.01 or later is required. However, if you're | +| | running on Data Center GPUs (formerly Tesla) such as T4, you can use | +| | any of the following NVIDIA Driver versions: | +| | | +| | * 418.40 (or later R418) | +| | * 440.33 (or later R440) | +| | * 450.51 (or later R450) | +| | * 460.27 (or later R460) | +| | | +| | **NOTE**: The CUDA Driver Compatibility Package doesn’t support all | +| | drivers. | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| GPU Model | * `NVIDIA Ampere GPU Architecture | +| | `_ | +| | * `Turning `_ | +| | * `Volta | +| | `_ | +| | * `Pascal | +| | `_ | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| **Base Container Image** | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| Container Operating System | Ubuntu version 20.04 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| Base Container | Triton version 21.07 | Triton version 21.10 | Triton version 21.11 | Triton version 21.12 | Triton version 22.02 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| CUDA | 11.6 | 11.6.55 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| RMM | 21.08 | 21.10 | 21.12 | 21.12 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| cuDF | 21.06 | 21.08 | 21.10 | 21.12 | 21.12.02 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| cuDNN | N/A | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| HugeCTR | 3.2 | 3.2.1 | 3.3 | 3.3.1 | 3.4 | 3.4.1 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| NVTabular | 0.7 | 0.7.1 | 0.8 | 0.9 | 0.10 | 0.11.0 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ +| SM | 60; 61; 70; 75; 80 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+----------------------+ :raw-html:`
` @@ -80,60 +80,60 @@ The following tables provide the software and model versions that NVTabular supp :raw-html:`

Table 2: Support matrix for the Merlin Training (merlin-training) container

` -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| | **Version 0.6** | **Version 21.09** | **Version 21.11** | **Version 21.12** | **Version 22.01** | **Version 22.02** | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| **DGX** | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| DGX System | * DGX-1 | -| | * DGX-2 | -| | * DGX A100 | -| | * DGX Station | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| Operating System | Ubuntu version 20.04 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| **NVIDIA Certified Systems** | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| NVIDIA Driver | NVIDIA Driver version 465.19.01 or later is required. However, if you're | -| | running on Data Center GPUs (formerly Tesla) such as T4, you can use | -| | any of the following NVIDIA Driver versions: | -| | | -| | * 418.40 (or later R418) | -| | * 440.33 (or later R440) | -| | * 450.51 (or later R450) | -| | * 460.27 (or later R460) | -| | | -| | **NOTE**: The CUDA Driver Compatibility Package doesn’t support all | -| | drivers. | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| GPU Model | * `NVIDIA Ampere GPU Architecture | -| | `_ | -| | * `Turning `_ | -| | * `Volta | -| | `_ | -| | * `Pascal | -| | `_ | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| **Base Container Image** | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| Container Operating System | Ubuntu version 20.04 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| Base Container | N/A | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| CUDA | 11.6 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| RMM | 21.06 | 21.08 | 21.10 | 21.12 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| cuDF | 21.06 | 21.08 | 21.10 | 21.12 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| cuDNN | 8.2.2 | 8.3.0 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| HugeCTR | 3.1 | 3.2 | 3.2.1 | 3.3 | 3.3.1 | 3.4 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| NVTabular | 0.6 | 0.7 | 0.7.1 | 0.8 | 0.9 | 0.10 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ -| SM | 60; 61; 70; 75; 80 | -+-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+ ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| | **Version 21.09** | **Version 21.11** | **Version 21.12** | **Version 22.01** | **Version 22.02** | **Version 22.03** | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| **DGX** | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| DGX System | * DGX-1 | +| | * DGX-2 | +| | * DGX A100 | +| | * DGX Station | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| Operating System | Ubuntu version 20.04 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| **NVIDIA Certified Systems** | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| NVIDIA Driver | NVIDIA Driver version 465.19.01 or later is required. However, if you're | +| | running on Data Center GPUs (formerly Tesla) such as T4, you can use | +| | any of the following NVIDIA Driver versions: | +| | | +| | * 418.40 (or later R418) | +| | * 440.33 (or later R440) | +| | * 450.51 (or later R450) | +| | * 460.27 (or later R460) | +| | | +| | **NOTE**: The CUDA Driver Compatibility Package doesn’t support all | +| | drivers. | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| GPU Model | * `NVIDIA Ampere GPU Architecture | +| | `_ | +| | * `Turning `_ | +| | * `Volta | +| | `_ | +| | * `Pascal | +| | `_ | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| **Base Container Image** | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| Container Operating System | Ubuntu version 20.04 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| Base Container | N/A | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| CUDA | 11.6 | 11.6.55 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| RMM | 21.08 | 21.10 | 21.12 | 21.12.0a0+31.g0acbd51 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| cuDF | 21.08 | 21.10 | 21.12 | 21.12.00a+293.g0930f712e6 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| cuDNN | 8.3.0 | 8.3.2 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| HugeCTR | 3.2 | 3.2.1 | 3.3 | 3.3.1 | 3.4 | 3.4.1 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| NVTabular | 0.7 | 0.7.1 | 0.8 | 0.9 | 0.10 | 0.11.0 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ +| SM | 60; 61; 70; 75; 80 | ++-----------------------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+-----------------------------------------+---------------------------+ :raw-html:`
` @@ -141,62 +141,62 @@ The following tables provide the software and model versions that NVTabular supp :raw-html:`

Table 3: Support matrix for the Merlin TensorFlow Training (merlin-tensorflow-training) container

` -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| | **Version 0.6** | **Version 21.09** | **Version 21.11** | **Version 21.12** | **Version 22.01** | **Version 22.02** | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| **DGX** | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| DGX System | * DGX-1 | -| | * DGX-2 | -| | * DGX A100 | -| | * DGX Station | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| Operating System | Ubuntu version 20.04 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| **NVIDIA Certified Systems** | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| NVIDIA Driver | NVIDIA Driver version 465.19.01 or later is required. However, if you're | -| | running on Data Center GPUs (formerly Tesla) such as T4, you can use | -| | any of the following NVIDIA Driver versions: | -| | | -| | * 418.40 (or later R418) | -| | * 440.33 (or later R440) | -| | * 450.51 (or later R450) | -| | * 460.27 (or later R460) | -| | | -| | **NOTE**: The CUDA Driver Compatibility Package doesn’t support all | -| | drivers. | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| GPU Model | * `NVIDIA Ampere GPU Architecture | -| | `_ | -| | * `Turning `_ | -| | * `Volta | -| | `_ | -| | * `Pascal | -| | `_ | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| **Base Container Image** | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| Container Operating System | Ubuntu version 20.04 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| Base Container | nvcr.io/nvidia/tensorflow:21.07-tf2-py3 | nvcr.io/nvidia/tensorflow:21.07-tf2-py3 | nvcr.io/nvidia/tensorflow:21.10-tf2-py3 | nvcr.io/nvidia/tensorflow:21.11-tf2-py3 | nvcr.io/nvidia/tensorflow:12.12-tf2-py3 | -| | | | | | | -| | \*Customized with TensorFlow version 2.4.2 | | | | | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| CUDA | 11.6 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| RMM | 21.06 | 21.08 | 21.10 | 21.12 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| cuDF | 21.06 | 21.08 | 21.10 | 21.12 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| cuDNN | N/A | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| HugeCTR | 3.1 | 3.2 | 3.2.1 | 3.3 | 3.3.1 | 3.4 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| NVTabular | 0.6 | 0.7 | 0.7.1 | 0.8 | 0.9 | 0.10 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| SM | 60; 61; 70; 75; 80 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| | **Version 21.09** | **Version 21.11** | **Version 21.12** | **Version 22.01** | **Version 22.02** | **Version 22.03** | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| **DGX** | | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| DGX System | * DGX-1 | +| | * DGX-2 | +| | * DGX A100 | +| | * DGX Station | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| Operating System | Ubuntu version 20.04 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| **NVIDIA Certified Systems** | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| NVIDIA Driver | NVIDIA Driver version 465.19.01 or later is required. However, if you're | +| | running on Data Center GPUs (formerly Tesla) such as T4, you can use | +| | any of the following NVIDIA Driver versions: | +| | | +| | * 418.40 (or later R418) | +| | * 440.33 (or later R440) | +| | * 450.51 (or later R450) | +| | * 460.27 (or later R460) | +| | | +| | **NOTE**: The CUDA Driver Compatibility Package doesn’t support all | +| | drivers. | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| GPU Model | * `NVIDIA Ampere GPU Architecture | +| | `_ | +| | * `Turning `_ | +| | * `Volta | +| | `_ | +| | * `Pascal | +| | `_ | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| **Base Container Image** | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| Container Operating System | Ubuntu version 20.04 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| Base Container | nvcr.io/nvidia/tensorflow:21.07-tf2-py3 | nvcr.io/nvidia/tensorflow:21.10-tf2-py3 | nvcr.io/nvidia/tensorflow:21.11-tf2-py3 | nvcr.io/nvidia/tensorflow:12.12-tf2-py3 | nvcr.io/nvidia/tensorflow:22.02-tf2-py3 | +| | | | | | | +| | \*Customized with TensorFlow version 2.4.2 | | | | | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| CUDA | 11.6 | 11.6.55 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| RMM | 21.08 | 21.10 | 21.12 | 21.12.0a0+31.g0acbd51 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| cuDF | 21.08 | 21.10 | 21.12 | 21.12.0a0+293.g0930f712e6 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| cuDNN | N/A | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| HugeCTR | 3.2 | 3.2.1 | 3.3 | 3.3.1 | 3.4 | 3.4.1 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| NVTabular | 0.7 | 0.7.1 | 0.8 | 0.9 | 0.10 | 0.11.0 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ +| SM | 60; 61; 70; 75; 80 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-----------------------------------------+ :raw-html:`
` @@ -204,58 +204,58 @@ The following tables provide the software and model versions that NVTabular supp :raw-html:`

Table 4: Support matrix for the Merlin PyTorch Training (merlin-pytorch-training) container

` -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| | **Version 0.6** | **Version 21.09** | **Version 21.11** | **Version 21.12** | **Version 22.01** | **Version 22.02** | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| **DGX** | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| DGX System | * DGX-1 | -| | * DGX-2 | -| | * DGX A100 | -| | * DGX Station | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| Operating System | Ubuntu version 20.04 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| **NVIDIA Certified Systems** | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| NVIDIA Driver | NVIDIA Driver version 465.19.01 or later is required. However, if you're | -| | running on Data Center GPUs (formerly Tesla) such as T4, you can use | -| | any of the following NVIDIA Driver versions: | -| | | -| | * 418.40 (or later R418) | -| | * 440.33 (or later R440) | -| | * 450.51 (or later R450) | -| | * 460.27 (or later R460) | -| | | -| | **NOTE**: The CUDA Driver Compatibility Package doesn’t support all | -| | drivers. | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| GPU Model | * `NVIDIA Ampere GPU Architecture | -| | `_ | -| | * `Turning `_ | -| | * `Volta | -| | `_ | -| | * `Pascal | -| | `_ | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| **Base Container Image** | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| Container Operating System | Ubuntu version 20.04 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| Base Container | nvcr.io/nvidia/pytorch:21.07-py3 | nvcr.io/nvidia/pytorch:21.10-py3 | nvcr.io/nvidia/pytorch:21.11-py3 | nvcr.io/nvidia/pytorch:21.12-py3 | -| | | | | | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| CUDA | 11.6 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| RMM | 21.06 | 21.08 | 21.10 | 21.12 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| cuDF | 21.06 | 21.08 | 21.10 | 21.12 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| cuDNN | N/A | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| HugeCTR | N/A | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| NVTabular | 0.6 | 0.7 | 0.7.1 | 0.8 | 0.9 | 0.10 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ -| SM | 60; 61; 70; 75; 80 | -+-----------------------------------------------------+------------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+ ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| | **Version 21.09** | **Version 21.11** | **Version 21.12** | **Version 22.01** | **Version 22.02** | **Version 22.03** | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| **DGX** | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| DGX System | * DGX-1 | +| | * DGX-2 | +| | * DGX A100 | +| | * DGX Station | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| Operating System | Ubuntu version 20.04 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| **NVIDIA Certified Systems** | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| NVIDIA Driver | NVIDIA Driver version 465.19.01 or later is required. However, if you're | +| | running on Data Center GPUs (formerly Tesla) such as T4, you can use | +| | any of the following NVIDIA Driver versions: | +| | | +| | * 418.40 (or later R418) | +| | * 440.33 (or later R440) | +| | * 450.51 (or later R450) | +| | * 460.27 (or later R460) | +| | | +| | **NOTE**: The CUDA Driver Compatibility Package doesn’t support all | +| | drivers. | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| GPU Model | * `NVIDIA Ampere GPU Architecture | +| | `_ | +| | * `Turning `_ | +| | * `Volta | +| | `_ | +| | * `Pascal | +| | `_ | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| **Base Container Image** | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| Container Operating System | Ubuntu version 20.04 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| Base Container | nvcr.io/nvidia/pytorch:21.07-py3 | nvcr.io/nvidia/pytorch:21.10-py3 | nvcr.io/nvidia/pytorch:21.11-py3 | nvcr.io/nvidia/pytorch:21.12-py3 | +| | | | | | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| CUDA | 11.6 | 11.6.55 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| RMM | 21.08 | 21.10 | 21.12 | 21.12.0a0+31.g0acbd51 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| cuDF | 21.08 | 21.10 | 21.12 | 21.12.0a0+293.g0930f712e6 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| cuDNN | N/A | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| HugeCTR | N/A | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| NVTabular | 0.7 | 0.7.1 | 0.8 | 0.9 | 0.10 | 0.11.0 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+ +| SM | 60; 61; 70; 75; 80 | ++-----------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+-----------------------------------------------------------+---------------------+-------------------------------------+