From 8a55cd5bf591708e6387655c615fe3a62f796f31 Mon Sep 17 00:00:00 2001
From: Miggi <noreply@mymiggi.de>
Date: Fri, 27 Jun 2025 14:30:47 +0200
Subject: [PATCH] Update torch to 2.7.1 & Cuda 12.8.1 in Docker

---
 docker/gpu/Dockerfile | 2 +-
 pyproject.toml        | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/docker/gpu/Dockerfile b/docker/gpu/Dockerfile
index 44c1ba7..5bc7b2e 100644
--- a/docker/gpu/Dockerfile
+++ b/docker/gpu/Dockerfile
@@ -1,4 +1,4 @@
-FROM --platform=$BUILDPLATFORM nvidia/cuda:12.8.0-cudnn-runtime-ubuntu24.04
+FROM --platform=$BUILDPLATFORM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04
 # Set non-interactive frontend
 ENV DEBIAN_FRONTEND=noninteractive
 
diff --git a/pyproject.toml b/pyproject.toml
index 5d082f7..ffbefe5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -44,10 +44,10 @@ dependencies = [
 
 [project.optional-dependencies]
 gpu = [
-    "torch==2.6.0+cu124",
+    "torch==2.7.1+cu128",
 ]
 cpu = [
-    "torch==2.6.0",
+    "torch==2.7.1",
 ]
 test = [
     "pytest==8.3.5",
@@ -79,7 +79,7 @@ explicit = true
 
 [[tool.uv.index]]
 name = "pytorch-cuda"
-url = "https://download.pytorch.org/whl/cu124"
+url = "https://download.pytorch.org/whl/cu128"
 explicit = true
 
 [build-system]