Spaces:
Configuration error
Configuration error
zero_gpu doesn't support pytorch 2.2
Browse files- Dockerfile +3 -3
- requirements.txt +2 -2
Dockerfile
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
FROM nvidia/cuda:12.
|
2 |
ENV DEBIAN_FRONTEND=noninteractive
|
3 |
RUN apt-get update && \
|
4 |
apt-get upgrade -y && \
|
@@ -34,13 +34,13 @@ WORKDIR ${HOME}/app
|
|
34 |
|
35 |
RUN curl https://pyenv.run | bash
|
36 |
ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
|
37 |
-
ARG PYTHON_VERSION=3.11
|
38 |
RUN pyenv install ${PYTHON_VERSION} && \
|
39 |
pyenv global ${PYTHON_VERSION} && \
|
40 |
pyenv rehash && \
|
41 |
pip install --no-cache-dir -U pip setuptools wheel
|
42 |
|
43 |
-
RUN pip install --no-cache-dir -U torch==2.
|
44 |
COPY --chown=1000 requirements.txt /tmp
|
45 |
RUN pip install --no-cache-dir -r /tmp/requirements.txt
|
46 |
|
|
|
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 && \
|
|
|
34 |
|
35 |
RUN curl https://pyenv.run | bash
|
36 |
ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
|
37 |
+
ARG PYTHON_VERSION=3.10.11
|
38 |
RUN pyenv install ${PYTHON_VERSION} && \
|
39 |
pyenv global ${PYTHON_VERSION} && \
|
40 |
pyenv rehash && \
|
41 |
pip install --no-cache-dir -U pip setuptools wheel
|
42 |
|
43 |
+
RUN pip install --no-cache-dir -U torch==2.1.1+cu121 torchvision>=0.16+cu121 --extra-index-url https://download.pytorch.org/whl/cu121
|
44 |
COPY --chown=1000 requirements.txt /tmp
|
45 |
RUN pip install --no-cache-dir -r /tmp/requirements.txt
|
46 |
|
requirements.txt
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
-
torch==2.2.1
|
2 |
-
pytorch-lightning==2.
|
3 |
torch-geometric==2.5.0
|
4 |
pytorch-cluster==1.6.3
|
5 |
scikit-learn-intelex==2024.1.0
|
|
|
1 |
+
#torch==2.2.1
|
2 |
+
pytorch-lightning==2.1.4
|
3 |
torch-geometric==2.5.0
|
4 |
pytorch-cluster==1.6.3
|
5 |
scikit-learn-intelex==2024.1.0
|