Browse Source
Update Dockerfile reorder installs (#6672)
Also `nvidia-tensorboard-plugin-dlprof`, `nvidia-tensorboard` are no longer installed in NVCR base.modifyDataloader
Glenn Jocher
GitHub
2 years ago
1 changed files with 2 additions and 2 deletions
+ 2
- 2
Dockerfile
View File
| # Install python dependencies | # Install python dependencies | ||||
| COPY requirements.txt . | COPY requirements.txt . | ||||
| RUN python -m pip install --upgrade pip | RUN python -m pip install --upgrade pip | ||||
| RUN pip uninstall -y nvidia-tensorboard nvidia-tensorboard-plugin-dlprof torch torchvision | |||||
| RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook | |||||
| RUN pip uninstall -y torch torchvision torchtext | |||||
| RUN pip install --no-cache torch==1.10.2+cu113 torchvision==0.11.3+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html | RUN pip install --no-cache torch==1.10.2+cu113 torchvision==0.11.3+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html | ||||
| RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook | |||||
| # RUN pip install --no-cache -U torch torchvision | # RUN pip install --no-cache -U torch torchvision | ||||
| # Create working directory | # Create working directory |
Loading…