diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile
index 79e7f40..65829a3 100644
--- a/.devcontainer/Dockerfile
+++ b/.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