Update Dockerfile

Dockerfile

@@ -1,7 +1,13 @@
-FROM python:3.10.12
+FROM nvidia/cuda:11.1.1-cudnn8-runtime-ubuntu20.04
 
+# Set up your working directory
 WORKDIR /code
 
+# Install any other dependencies you need
+# For example, let's say you want to install Python and pip
+RUN apt-get update && \
+    apt-get install -y python3 python3-pip
+    
 COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt