доработка для запуска со скрипта на хосте на ROCm

2025-09-07 16:58:55 +03:00
parent 50fcb2d025
commit b212d83161
2 changed files with 16 additions and 0 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,9 @@
+# PyTorch с поддержкой ROCm
+--index-url https://download.pytorch.org/whl/rocm6.0
+torch
+torchaudio
+
+# Остальные зависимости
 fastapi
 uvicorn[standard]
 python-multipart