Spaces:
Build error
Build error
Update Dockerfile with compatible torch version
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -43,7 +43,7 @@ RUN pyenv install ${PYTHON_VERSION} && \
|
|
| 43 |
pyenv rehash && \
|
| 44 |
pip install --no-cache-dir -U pip setuptools wheel
|
| 45 |
|
| 46 |
-
RUN pip install --no-cache-dir -U torch==
|
| 47 |
COPY --chown=1000 requirements.txt /tmp/requirements.txt
|
| 48 |
RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
|
| 49 |
|
|
@@ -56,4 +56,4 @@ ENV PYTHONPATH=${HOME}/app \
|
|
| 56 |
GRADIO_SERVER_NAME=0.0.0.0 \
|
| 57 |
GRADIO_THEME=huggingface \
|
| 58 |
SYSTEM=spaces
|
| 59 |
-
CMD ["python", "app.py"]
|
|
|
|
| 43 |
pyenv rehash && \
|
| 44 |
pip install --no-cache-dir -U pip setuptools wheel
|
| 45 |
|
| 46 |
+
RUN pip install --no-cache-dir -U torch==2.0.1 torchvision==0.15.2
|
| 47 |
COPY --chown=1000 requirements.txt /tmp/requirements.txt
|
| 48 |
RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
|
| 49 |
|
|
|
|
| 56 |
GRADIO_SERVER_NAME=0.0.0.0 \
|
| 57 |
GRADIO_THEME=huggingface \
|
| 58 |
SYSTEM=spaces
|
| 59 |
+
CMD ["python", "app.py"]
|