Spaces:
Runtime error
Runtime error
Update torch version
Browse files- Dockerfile +1 -1
- requirements.txt +2 -2
Dockerfile
CHANGED
|
@@ -41,7 +41,7 @@ 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==1.
|
| 45 |
COPY --chown=1000 requirements.txt /tmp/requirements.txt
|
| 46 |
RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
|
| 47 |
|
|
|
|
| 41 |
pyenv rehash && \
|
| 42 |
pip install --no-cache-dir -U pip setuptools wheel
|
| 43 |
|
| 44 |
+
RUN pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1
|
| 45 |
COPY --chown=1000 requirements.txt /tmp/requirements.txt
|
| 46 |
RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
|
| 47 |
|
requirements.txt
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
-
torch==1.
|
| 2 |
-
torchvision==0.
|
| 3 |
accelerate==0.12.0
|
| 4 |
absl-py
|
| 5 |
ml_collections
|
|
|
|
| 1 |
+
torch==1.13.1
|
| 2 |
+
torchvision==0.14.1
|
| 3 |
accelerate==0.12.0
|
| 4 |
absl-py
|
| 5 |
ml_collections
|