From 50fcb2d0252a454b3a630abe989dd343f16fec7d Mon Sep 17 00:00:00 2001
From: vladislav <nauka.2.0.vs@gmail.com>
Date: Sun, 7 Sep 2025 12:23:31 +0300
Subject: [PATCH] =?UTF-8?q?fix:=20=D0=BD=D0=B5=D0=B2=D0=B5=D1=80=D0=BD?=
 =?UTF-8?q?=D1=8B=D0=B9=20=D0=B1=D0=B0=D0=B7=D0=BE=D0=B2=D1=8B=D0=B9=20?=
 =?UTF-8?q?=D0=BE=D0=B1=D1=80=D0=B0=D0=B7=20=D0=B4=D0=BB=D1=8F=20ROCm?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Dockerfile | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 3504e99..8bacee9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,8 @@
-# Use official Python image as base
-FROM python:3.10-slim
+# NVIDIA
+#FROM python:3.10-slim
+
+# AMD
+FROM rocm/pytorch:latest
 
 # Set working directory
 WORKDIR /app