From cca3e3f3d172c11ecd84a483c09845c1e2c3e658 Mon Sep 17 00:00:00 2001
From: matt-personal <mattmendivil49@gmail.com>
Date: Sat, 31 Jan 2026 14:59:13 -0500
Subject: [PATCH] update docker build

---
 .github/workflows/docker.yml | 22 ++++++++++++++++++++--
 Dockerfile                   |  2 +-
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index 265599a..a881117 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -3,9 +3,28 @@ name: Build and Push Docker Image
 on:
   push:
     branches: [main]
+  pull_request:
+    branches: [main]
 
 jobs:
-  build:
+  test-build:
+    if: github.event_name == 'pull_request'
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Test Docker build
+        uses: docker/build-push-action@v5
+        with:
+          context: .
+          push: false
+          tags: ghcr.io/${{ github.repository }}:test
+
+  build-and-push:
+    if: github.event_name == 'push'
     runs-on: ubuntu-latest
     permissions:
       contents: read
@@ -27,4 +46,3 @@ jobs:
           context: .
           push: true
           tags: ghcr.io/${{ github.repository }}:latest
-          
\ No newline at end of file
diff --git a/Dockerfile b/Dockerfile
index d79c56c..9b004dc 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM runpod/pytorch:2.4.1-py3.11-cuda12.4.1-devel-ubuntu22.04
+FROM runpod/pytorch:2.8.0-py3.11-cuda12.8.1-cudnn-devel-ubuntu22.04
 
 WORKDIR /app