fixed a lot of things and actually got it to work
This commit is contained in:
@@ -1,5 +1,9 @@
|
||||
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 \
|
||||
@@ -7,7 +11,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
COPY requirements.txt .
|
||||
RUN pip install --no-cache-dir -r 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 .
|
||||
|
||||
Reference in New Issue
Block a user