Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
+FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
 #FROM nvidia/cuda:11.6.1-devel-ubuntu20.04
 
@@ -10,7 +10,7 @@ RUN apt-get update
 #RUN apt-get -y update && apt-get -y install software-properties-common \
 #&& add-apt-repository ppa:deadsnakes/ppa && apt install -y python3.10
 
-RUN apt-get install -y python3 python3-pip
+RUN apt-get install -y python3 python3-pip git
 WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt