akseljoonas HF Staff commited on
Commit
ccd825d
·
verified ·
1 Parent(s): 432f2c4

Update Dockerfile with compatible torch version

Browse files
Files changed (1) hide show
  1. 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==1.12.1 torchvision==0.13.1
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"]