FROM nvcr.io/nvidia/pytorch:22.06-py3 (#8600)
Browse files- utils/docker/Dockerfile +2 -2
utils/docker/Dockerfile
CHANGED
|
@@ -3,7 +3,7 @@
|
|
| 3 |
# Image is CUDA-optimized for YOLOv5 single/multi-GPU training and inference
|
| 4 |
|
| 5 |
# Start FROM NVIDIA PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
|
| 6 |
-
FROM nvcr.io/nvidia/pytorch:22.
|
| 7 |
RUN rm -rf /opt/pytorch # remove 1.2GB dir
|
| 8 |
|
| 9 |
# Downloads to user config dir
|
|
@@ -15,7 +15,7 @@ RUN apt update && apt install --no-install-recommends -y zip htop screen libgl1-
|
|
| 15 |
# Install pip packages
|
| 16 |
COPY requirements.txt .
|
| 17 |
RUN python -m pip install --upgrade pip wheel
|
| 18 |
-
RUN pip uninstall -y torch torchvision
|
| 19 |
RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook Pillow>=9.1.0 \
|
| 20 |
'opencv-python<4.6.0.66' \
|
| 21 |
--extra-index-url https://download.pytorch.org/whl/cu113
|
|
|
|
| 3 |
# Image is CUDA-optimized for YOLOv5 single/multi-GPU training and inference
|
| 4 |
|
| 5 |
# Start FROM NVIDIA PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
|
| 6 |
+
FROM nvcr.io/nvidia/pytorch:22.06-py3
|
| 7 |
RUN rm -rf /opt/pytorch # remove 1.2GB dir
|
| 8 |
|
| 9 |
# Downloads to user config dir
|
|
|
|
| 15 |
# Install pip packages
|
| 16 |
COPY requirements.txt .
|
| 17 |
RUN python -m pip install --upgrade pip wheel
|
| 18 |
+
RUN pip uninstall -y Pillow torchtext # torch torchvision
|
| 19 |
RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook Pillow>=9.1.0 \
|
| 20 |
'opencv-python<4.6.0.66' \
|
| 21 |
--extra-index-url https://download.pytorch.org/whl/cu113
|