23 lines
549 B
Docker
23 lines
549 B
Docker
FROM rocm/pytorch:rocm6.4.1_ubuntu22.04_py3.10_pytorch_release_2.6.0
|
|
|
|
WORKDIR /app
|
|
|
|
RUN apt-get update && apt-get install -y \
|
|
ffmpeg \
|
|
python3-pip \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
COPY requirements.txt .
|
|
RUN pip install --no-cache-dir --default-timeout=100 -r requirements.txt
|
|
|
|
COPY . .
|
|
|
|
EXPOSE 9854
|
|
|
|
# Устанавливаем переменные окружения для ROCm
|
|
ENV HSA_OVERRIDE_GFX_VERSION=10.3.0
|
|
ENV PYTORCH_ROCM_ARCH=gfx1030
|
|
|
|
# Команда для запуска приложения
|
|
CMD ["python3", "app.py"]
|