From b151bedd4a27b23e716118c91af1846b188fd00a Mon Sep 17 00:00:00 2001
From: Vladimir Protsenko <protsenkovi@ssau.ru>
Date: Mon, 29 Sep 2025 20:06:07 +0400
Subject: [PATCH] updated pytorch image

---
 Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 3877057..62ac284 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,5 @@
-FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.8.0-cuda12.9-cudnn9-runtime
+# to compile with CUDA toolkit use pytorch/pytorch:2.8.0-cuda12.9-cudnn9-devel
 ENV TZ=Europe/Samara
 RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone