zideliu commited on
Commit
7da0921
1 Parent(s): 1ac4abf

Update torch version

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. 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.12.1 torchvision==0.13.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.12.1
2
- torchvision==0.13.1
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