Problem on NVIDIA CUDa Drivers

[FelipeSBarros]

Dear colleagues, I understand that the problem I am having is related to my computer configuration (NVIDIA drivers) and not with raster-vision.
Despite that I am asking here, to see if any one here has any idea about how I could fix it.

I am trying to run the quickstart process using the docker image ( pytorch-latest).
After creating the code and output folders, I run the code to access de docker console:

    -v ${RV_QUICKSTART_CODE_DIR}:/opt/src/code  \
    -v ${RV_QUICKSTART_OUT_DIR}:/opt/data/output \
    quay.io/azavea/raster-vision:pytorch-0.20 /bin/bash

And it shows me a warning:

==========
== CUDA ==
==========

CUDA Version 11.3.1

Container image Copyright (c) 2016-2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.

This container image and its contents are governed by the NVIDIA Deep Learning Container License.
By pulling and using the container, you accept the terms and conditions of this license:
https://developer.nvidia.com/ngc/nvidia-deep-learning-container-license

A copy of this license is made available in this container at /NGC-DL-CONTAINER-LICENSE for your convenience.

WARNING: The NVIDIA Driver was not detected.  GPU functionality will not be available.
   Use the NVIDIA Container Toolkit to start this container with GPU support; see
   https://docs.nvidia.com/datacenter/cloud-native/ .

I am using an notebook lenovo ideapad 3i with NVIDIA RTX3050:
The output of nvidia-smi givesme:

Tue Feb 14 22:14:04 2023       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 525.78.01    Driver Version: 525.78.01    CUDA Version: 12.0     |
|-------------------------------+----------------------+----------------------+
| 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  NVIDIA GeForce ...  Off  | 00000000:01:00.0 Off |                  N/A |
| N/A   45C    P3    N/A /  30W |      5MiB /  4096MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1902      G   /usr/lib/xorg/Xorg                  4MiB |
+-----------------------------------------------------------------------------+

If you have any clue about what is happening and how could I setup the drivers to have GPU functionality running, please, let me know

[AdeelH]

This might help: https://docs.rastervision.io/en/stable/setup/gpu.html#gpus-and-docker.

Make sure you have nvidia-docker installed. Also add --gpus=all (or --runtime=nvidia, if using an older docker version) to the docker run command. Or, if using docker/run (i.e. this script), pass the --gpu flag to it.

[FelipeSBarros]

Thanks @AdeelH ! As soon as possible I will came with an update. Best regards