WARNING: Detected NVIDIA A100-PCIE-40GB GPU, which is not yet supported in this version of the container

[Jhaprince]

Hi @tjyuyao, @chrisman1015,
I am also facing issues similar to this
#160 (comment),

It would be great help for me if you all please help me out with this.

These are the exact error message:

=============
== PyTorch ==

NVIDIA Release 19.05 (build 6411784)
PyTorch Version 1.1.0a0+828a6a3

Container image Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.

Copyright (c) 2014-2019 Facebook Inc.
Copyright (c) 2011-2014 Idiap Research Institute (Ronan Collobert)
Copyright (c) 2012-2014 Deepmind Technologies (Koray Kavukcuoglu)
Copyright (c) 2011-2012 NEC Laboratories America (Koray Kavukcuoglu)
Copyright (c) 2011-2013 NYU (Clement Farabet)
Copyright (c) 2006-2010 NEC Laboratories America (Ronan Collobert, Leon Bottou, Iain Melvin, Jason Weston)
Copyright (c) 2006 Idiap Research Institute (Samy Bengio)
Copyright (c) 2001-2004 Idiap Research Institute (Ronan Collobert, Samy Bengio, Johnny Mariethoz)
Copyright (c) 2015 Google Inc.
Copyright (c) 2015 Yangqing Jia
Copyright (c) 2013-2016 The Caffe contributors
All rights reserved.

Various files include modifications (c) NVIDIA CORPORATION. All rights reserved.
NVIDIA modifications are covered by the license terms that apply to the underlying project or file.
WARNING: Detected NVIDIA A100-PCIE-40GB GPU, which is not yet supported in this version of the container
WARNING: Detected NVIDIA A100-PCIE-40GB GPU, which is not yet supported in this version of the container
WARNING: Detected NVIDIA A100-PCIE-40GB GPU, which is not yet supported in this version of the container
WARNING: Detected NVIDIA A100-PCIE-40GB GPU, which is not yet supported in this version of the container
WARNING: Detected NVIDIA A100-PCIE-40GB GPU, which is not yet supported in this version of the container

###Here I am also giving you the details of nvidia drivers and cuda

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.91.03 Driver Version: 460.91.03 CUDA Version: 11.2 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 A100-PCIE-40GB Off | 00000000:01:00.0 Off | 0 |
| N/A 42C P0 42W / 250W | 29452MiB / 40536MiB | 0% Default |
| | | Disabled |
+-------------------------------+----------------------+----------------------+
| 1 A100-PCIE-40GB Off | 00000000:41:00.0 Off | 0 |
| N/A 58C P0 94W / 250W | 36248MiB / 40536MiB | 28% Default |
| | | Disabled |
+-------------------------------+----------------------+----------------------+
| 2 A100-PCIE-40GB Off | 00000000:81:00.0 Off | 0 |
| N/A 53C P0 81W / 250W | 31684MiB / 40536MiB | 42% Default |
| | | Disabled |
+-------------------------------+----------------------+----------------------+
| 3 A100-PCIE-40GB Off | 00000000:C1:00.0 Off | 0 |
| N/A 39C P0 36W / 250W | 3MiB / 40536MiB | 0% Default |
| | | Disabled |
+-------------------------------+----------------------+----------------------+
| 4 A100-PCIE-40GB Off | 00000000:E1:00.0 Off | 0 |
| N/A 41C P0 42W / 250W | 6251MiB / 40536MiB | 10% Default |
| | | Disabled |
+-------------------------------+----------------------+----------------------+

My docker version is Docker version 20.10.7, build f0df350

I can also provide other required details if in future required.
Hoping to receive a quick response from your side

Thank you in advance