perler commited on
Commit
60bec67
1 Parent(s): f5c1ca5

kaolin supports only up to torch 2.1.1

Browse files
Files changed (3) hide show
  1. Dockerfile +2 -2
  2. README.md +1 -1
  3. requirements.txt +1 -1
Dockerfile CHANGED
@@ -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==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
 
41
  pyenv rehash && \
42
  pip install --no-cache-dir -U pip setuptools wheel
43
 
44
+ RUN pip install --no-cache-dir -U torch==2.1.1+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.1.1_cu121.html
48
  RUN pip install --no-cache-dir -U gradio==3.34.0
49
 
50
  COPY --chown=1000 . ${HOME}/app
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: Ppsurf
3
  emoji: 🏃
4
  colorFrom: green
5
  colorTo: gray
 
1
  ---
2
+ title: PPSurf
3
  emoji: 🏃
4
  colorFrom: green
5
  colorTo: gray
requirements.txt CHANGED
@@ -8,7 +8,7 @@ matplotlib==3.7.1
8
  ninja==1.11.1
9
  opencv-python-headless==4.7.0.72
10
  pyrallis==0.3.1
11
- torch==2.2.0
12
  torchvision==0.17.0
13
  tqdm==4.65.0
14
  transformers==4.29.1
 
8
  ninja==1.11.1
9
  opencv-python-headless==4.7.0.72
10
  pyrallis==0.3.1
11
+ torch==2.1.1
12
  torchvision==0.17.0
13
  tqdm==4.65.0
14
  transformers==4.29.1