Spaces:
Configuration error
Configuration error
kaolin supports only up to torch 2.1.1
Browse files- Dockerfile +2 -2
- README.md +1 -1
- 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.
|
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.
|
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:
|
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.
|
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
|