Tensorflow Example Nvidia driver not found warning

Following the examples here :https://docs.nvidia.com/cuda/wsl-user-guide/index.html

n-body example works fine (output as shown in the blog), but the Tensorflow ResNet-50 examples gives the following warning.

WARNING: The NVIDIA Driver was not detected. GPU functionality will not be available.

I am not sure what the issue is here, any suggestions on what the issue may be ?

This should be a minor issue with the startup script of these containers. It checks for some Driver file that are specific to Native Linux to display this message. The GPU should still be available despite this warning message.

Let us know if you hit any other issues,

Thanks !

Hi,
Thanks for the clarification.

I tried running the example, but it gives the following error:

tensorflow/stream_executor/cuda/cuda_driver.cc:801] failed to allocate 534.62M (560594944 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY: out of memory

My device is an RTX 2060 6GB.