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"]
|