From c22827ebda14be1464d50b6bc162b07973bb0257 Mon Sep 17 00:00:00 2001
From: Bea Steers <bea.steers@gmail.com>
Date: Tue, 8 Oct 2024 16:30:20 -0400
Subject: [PATCH] Update Dockerfile.gpu

---
 Dockerfile.gpu | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index 755975f..ddebe54 100644
--- 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"