27 lines
728 B
Docker
27 lines
728 B
Docker
FROM rocm/pytorch:rocm7.1.1_ubuntu22.04_py3.11_pytorch_release_2.10.0
|
|
|
|
# MIOpen optimizations: cache convolution solver solutions instead of re-solving on every inference
|
|
ENV MIOPEN_FIND_MODE=1
|
|
ENV MIOPEN_DEBUG_CONV_GEMM=0
|
|
|
|
WORKDIR /app
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
build-essential \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
COPY requirements.txt .
|
|
RUN pip install --no-cache-dir -r requirements.txt && \
|
|
python -c "import nltk; nltk.download('averaged_perceptron_tagger_eng'); nltk.download('cmudict')"
|
|
|
|
COPY wyoming_glados/ ./wyoming_glados/
|
|
COPY download_model.py .
|
|
COPY entrypoint.sh .
|
|
RUN chmod +x entrypoint.sh
|
|
|
|
RUN mkdir -p /data
|
|
|
|
EXPOSE 10200
|
|
|
|
ENTRYPOINT ["/app/entrypoint.sh"]
|