Merge pull request #65 from d-kleine/main

Updated Dockerfile

Author: rasbt

.devcontainer/Dockerfile

@@ -1,10 +1,9 @@
-FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
-RUN <<EOF
-apt-get update
-apt-get install -y --no-install-recommends git
-rm -rf /var/lib/apt/lists/*
-EOF
+RUN apt-get update && \
+    apt-get upgrade -y && \
+    apt-get install -y git && \
+    rm -rf /var/lib/apt/lists/*
 
 COPY requirements.txt requirements.txt