33 lines
934 B
Docker
33 lines
934 B
Docker
# ROCm-совместимый образ с PyTorch
|
|
FROM rocm/pytorch:rocm5.7.3_ubuntu22.04_py3.10_pytorch2.1.2
|
|
|
|
# Установка системных зависимостей
|
|
RUN apt-get update && apt-get install -y \
|
|
ffmpeg \
|
|
curl \
|
|
git \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Создание рабочей директории
|
|
WORKDIR /app
|
|
|
|
# Установка переменных окружения для Python
|
|
ENV PYTHONUNBUFFERED=1
|
|
ENV HSA_OVERRIDE_GFX_VERSION=10.3.0
|
|
|
|
# Установка Python зависимостей (без torch/torchaudio - уже в базовом образе)
|
|
RUN pip install --no-cache-dir \
|
|
fastapi==0.109.0 \
|
|
uvicorn[standard]==0.27.0 \
|
|
python-multipart==0.0.6 \
|
|
openai-whisper==20231117
|
|
|
|
# Копирование исходного кода
|
|
COPY main.py .
|
|
|
|
# Экспорт порта
|
|
EXPOSE 8080
|
|
|
|
# Запуск приложения
|
|
CMD ["python", "main.py"]
|