fix: неверный базовый образ для ROCm

2025-09-07 12:23:31 +03:00
parent 1f13bc4f86
commit 50fcb2d025
1 changed files with 5 additions and 2 deletions
--- a/7
+++ b/7
@@ -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