Update Dockerfile.gpu

VIDA-NYU · Oct 8, 2024 · c22827e · c22827e
1 parent 1153d55
commit c22827e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile.gpu b/Dockerfile.gpu
@@ -1,6 +1,6 @@
 FROM pytorch/pytorch:2.4.0-cuda11.8-cudnn9-runtime
-RUN rm /etc/apt/sources.list.d/cuda.list
-RUN rm /etc/apt/sources.list.d/nvidia-ml.list
+#RUN rm /etc/apt/sources.list.d/cuda.list
+#RUN rm /etc/apt/sources.list.d/nvidia-ml.list
 ARG DEBIAN_FRONTEND=noninteractive
 ENV TZ="America/New_York"