NGC container - What GPU's

Accelerated Computing NGC GPU Cloud General
1

I am trying to find a list of GPU’s supported by the following NGC container. TensorFlow | NVIDIA NGC

It must be somewhere right? Can anyone point me to it?

I found this: ERROR: Detected NVIDIA Tesla K80 GPU, which is not supported by this container

which says:

The GPU-accelerated deep learning containers are tuned, tested, and certified by NVIDIA to run on NVIDIA TITAN V, TITAN Xp, TITAN X (Pascal), NVIDIA Quadro GV100, GP100 and P6000, NVIDIA DGX Systems​.

Please see https://www.nvidia.com/en-us/gpu-cloud/deep-learning-containers/ for more information.

Sorry,
I don’t know how to figure out whether each GPU fits in this list.

2

This page TensorFlow Release 23.01 - NVIDIA Docs says:

GPU Requirements

Release 23.01 supports CUDA compute capability 6.0 and later. This corresponds to GPUs in the NVIDIA Pascal, NVIDIA Volta™, NVIDIA Turing™, NVIDIA Ampere architecture, and NVIDIA Hopper™ architecture families. For a list of GPUs to which this compute capability corresponds, see CUDA GPUs. For additional support details, see Deep Learning Frameworks Support Matrix.

This page: https://www.h3platform.com/blog-detail/22#:~:text=The%20RTX%204090%20and%20RTX,to%20Ampere’s%20Samsung%208nm%20process. Says.

The RTX 4090 and RTX A6000 Ada are based on Nvidia’s Ada Lovelace architecture, but the RTX 3090 and RTX A6000 are on the last generation Ampere architecture. In addition, the Ada Lovelace GPU adopts TSMC’s 4nm manufacturing process compared to Ampere’s Samsung 8nm process.

If the 4090 is Ada architecture, then it doesn’t seem to be on the list of GPUs supported by NGC considering the release note above. Yet, I have heard in more than one place that NGC does support 4090.

I can run model training on NGC on paperspace with the ampere A6000.

I don’t want to buy a 4090 physical machine only to find out I can’t get it to run my TF1 code.

This is confusing to me.

Help?
Signed - Confused.