Update Dockerfile

Dockerfile

@@ -1,7 +1,10 @@
 FROM python:3.9
 WORKDIR /app
 
-RUN apt-get install nvidia-container-runtime
+RUN distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
+  && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
+  && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list \
+  && apt-get update && sudo apt-get install -y nvidia-container-toolkit
 
 COPY ./requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt