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