From 1980c5f23604dedfebfb40a7627b74e730ae9ab8 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Fri, 28 Jun 2024 18:15:33 -0700
Subject: [PATCH] Use an image with earlier drivers

---
 .github/workflows/test.yml | 2 +-
 Dockerfile                 | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 26d7630d8..0a2ec95b4 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -23,7 +23,7 @@ jobs:
           - labels: [self-hosted, Linux, X64, rocm-7600-8gb]
             container: "rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2"
           - labels: [self-hosted, Linux, X64, cuda-3060-12gb]
-            container: "nvcr.io/nvidia/pytorch:24.04-py3"
+            container: "nvcr.io/nvidia/pytorch:24.03-py3"
     steps:
       - run: |
           apt update
diff --git a/Dockerfile b/Dockerfile
index 7f1a4fe72..eec56003a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/pytorch:24.04-py3
+FROM nvcr.io/nvidia/pytorch:24.03-py3
 RUN pip install --no-cache --no-build-isolation git+https://github.com/hiddenswitch/ComfyUI.git
 EXPOSE 8188
 WORKDIR /workspace