perler commited on
Commit
b58ba92
1 Parent(s): cef9302

space seems to require torch==2.2.0, trying to update TEXTure requirements

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
2
  ENV DEBIAN_FRONTEND=noninteractive
3
  RUN apt-get update && \
4
  apt-get upgrade -y && \
@@ -41,10 +41,10 @@ RUN pyenv install ${PYTHON_VERSION} && \
41
  pyenv rehash && \
42
  pip install --no-cache-dir -U pip setuptools wheel
43
 
44
- RUN pip install --no-cache-dir -U torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113
45
  COPY --chown=1000 requirements.txt /tmp
46
  RUN pip install --no-cache-dir -r /tmp/requirements.txt
47
- RUN pip install --no-cache-dir -U kaolin==0.13.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-1.12.1_cu113.html
48
  RUN pip install --no-cache-dir -U gradio==3.34.0
49
 
50
  COPY --chown=1000 . ${HOME}/app
 
1
+ FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
2
  ENV DEBIAN_FRONTEND=noninteractive
3
  RUN apt-get update && \
4
  apt-get upgrade -y && \
 
41
  pyenv rehash && \
42
  pip install --no-cache-dir -U pip setuptools wheel
43
 
44
+ RUN pip install --no-cache-dir -U torch==2.2.0+cu121 torchvision==0.17.0+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
45
  COPY --chown=1000 requirements.txt /tmp
46
  RUN pip install --no-cache-dir -r /tmp/requirements.txt
47
+ RUN pip install --no-cache-dir -U kaolin==0.15.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.2.0_cu121.html
48
  RUN pip install --no-cache-dir -U gradio==3.34.0
49
 
50
  COPY --chown=1000 . ${HOME}/app