Issue getting GPU working with docker image in ubuntu

[ashwinjoseph95]

Hi @koide3 ,
I think this work is really great!
I am trying to run the calibration on a similar environemnt (ROS1 and Ubuntu 20)
I have setup the docker image provided by you. I was able to do the preprocess step but when it came to next step for manually selcting points I realized there was a lot of lag and also that the docker was not using gpu even though I have provided the flags for it to do so.
My commands are as follows:

# Define paths
bag_path=$(realpath /home/knadmin/Ashwin/Targetless_calib/livox_ros1)
preprocessed_path=$(realpath /home/knadmin/Ashwin/Targetless_calib/livox_ros1_preprocessed)

docker run \
  -it \
  --net host \
  --runtime=nvidia --gpus all  \
  -e DISPLAY=$DISPLAY \
  -v $HOME/.Xauthority:/root/.Xauthority \
  -v $bag_path:/tmp/input_bags \
  -v $preprocessed_path:/tmp/preprocessed \
  -v pip_cache:/root/.cache/pip \
  koide3/direct_visual_lidar_calibration:noetic /bin/bash nvidia-smi

but apparenty nvidia-smi is not getting recognized. I have insalled all the nvidia and cuda dependencies in the local system.

I tried another example here with ubuntu docker iamge it detects the nvidia drivers:

What should i do to rectify this issue? I understand this might be slightly off topic but I would be really helpful to udnerstand what is going wrong and how I can fix it.