Spaces:
Paused
Paused
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
|