From 1019666122d3c1bec8508a2f03afeb79ac7fd646 Mon Sep 17 00:00:00 2001
From: Vladimir Protsenko <protsenkovi@ssau.ru>
Date: Tue, 13 Feb 2024 10:18:19 +0400
Subject: [PATCH] Updated version of the base image

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

diff --git a/Dockerfile b/Dockerfile
index 640026c..3877057 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
+FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
 ENV TZ=Europe/Samara
 RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone