glenn-jocher commited on
Commit
c6b4f84
1 Parent(s): b94b59e

Update Dockerfile `torch==1.11.0+cu113` (#6954)

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -11,7 +11,7 @@ COPY requirements.txt .
11
  RUN python -m pip install --upgrade pip
12
  RUN pip uninstall -y torch torchvision torchtext
13
  RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook \
14
- torch==1.10.2+cu113 torchvision==0.11.3+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
15
  # RUN pip install --no-cache -U torch torchvision
16
 
17
  # Create working directory
 
11
  RUN python -m pip install --upgrade pip
12
  RUN pip uninstall -y torch torchvision torchtext
13
  RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook \
14
+ torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
15
  # RUN pip install --no-cache -U torch torchvision
16
 
17
  # Create working directory