Spaces:
Runtime error
Runtime error
space seems to require torch==2.2.0, trying to update TEXTure requirements
Browse files- Dockerfile +3 -3
Dockerfile
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
FROM nvidia/cuda:
|
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==
|
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.
|
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
|