From f3221b3541eb2f2ae0beb67402ad46804257bfd6 Mon Sep 17 00:00:00 2001
From: hodanov <1031hoda@gmail.com>
Date: Sat, 8 Feb 2025 22:55:12 +0900
Subject: [PATCH] Update pytorch.

---
 app/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/app/Dockerfile b/app/Dockerfile
index a2051f5..4b3045b 100644
--- a/app/Dockerfile
+++ b/app/Dockerfile
@@ -5,4 +5,4 @@ RUN apt-get update \
     && apt-get autoremove -y \
     && apt-get clean -y \
     && rm -rf /var/lib/apt/lists/* \
-    && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu121 --no-cache-dir
\ No newline at end of file
+    && pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu126 --no-cache-dir
\ No newline at end of file