From a5650909a701c97a2e1539d73cb36a0236fbcd04 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 13:37:20 -0700 Subject: [PATCH 01/18] Update --- utils/docker/Dockerfile | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/utils/docker/Dockerfile b/utils/docker/Dockerfile index 9bb24bb6bf3e..97772a88c693 100644 --- a/utils/docker/Dockerfile +++ b/utils/docker/Dockerfile @@ -1,7 +1,7 @@ # YOLOv5 🚀 by Ultralytics, GPL-3.0 license # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch -FROM nvcr.io/nvidia/pytorch:21.10-py3 +FROM nvcr.io/nvidia/pytorch:22.04-py3 # Install linux packages RUN apt update && apt install -y zip htop screen libgl1-mesa-glx @@ -11,7 +11,7 @@ COPY requirements.txt . RUN python -m pip install --upgrade pip RUN pip uninstall -y torch torchvision torchtext RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook \ - torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html + torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 \ # RUN pip install --no-cache -U torch torchvision # Create working directory @@ -32,7 +32,7 @@ ENV OMP_NUM_THREADS=8 # Usage Examples ------------------------------------------------------------------------------------------------------- # Build and Push -# t=ultralytics/yolov5:latest && sudo docker build -f utils/docker/Dockerfile -t $t . && sudo docker push $t +# t=ultralytics/yolov5:latest_2204_torchinstall && sudo docker build -f utils/docker/Dockerfile -t $t . && sudo docker push $t # Pull and Run # t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all $t From f7e6e9f8d7292c1e524e602707c96c2b4de90f3a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 13:38:34 -0700 Subject: [PATCH 02/18] Update --- utils/docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/docker/Dockerfile b/utils/docker/Dockerfile index 97772a88c693..44427e4faa84 100644 --- a/utils/docker/Dockerfile +++ b/utils/docker/Dockerfile @@ -11,7 +11,7 @@ COPY requirements.txt . RUN python -m pip install --upgrade pip RUN pip uninstall -y torch torchvision torchtext RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook \ - torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 \ + torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 # RUN pip install --no-cache -U torch torchvision # Create working directory From 2dd1c7332e4e54bd02f0dec45f33298917cb9c3c Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 13:40:45 -0700 Subject: [PATCH 03/18] Do not install torch --- utils/docker/Dockerfile | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/utils/docker/Dockerfile b/utils/docker/Dockerfile index 44427e4faa84..128becb39a2f 100644 --- a/utils/docker/Dockerfile +++ b/utils/docker/Dockerfile @@ -9,9 +9,9 @@ RUN apt update && apt install -y zip htop screen libgl1-mesa-glx # Install python dependencies COPY requirements.txt . RUN python -m pip install --upgrade pip -RUN pip uninstall -y torch torchvision torchtext -RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook \ - torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 +RUN pip uninstall -y torchtext +RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook + # torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 # RUN pip install --no-cache -U torch torchvision # Create working directory From bd8c619a5649dd287cec2d47a788a4c4f2cf7cfb Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 19:01:21 -0700 Subject: [PATCH 04/18] Pillow>9.1 --- utils/docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/docker/Dockerfile b/utils/docker/Dockerfile index 128becb39a2f..53fa43105563 100644 --- a/utils/docker/Dockerfile +++ b/utils/docker/Dockerfile @@ -10,7 +10,7 @@ RUN apt update && apt install -y zip htop screen libgl1-mesa-glx COPY requirements.txt . RUN python -m pip install --upgrade pip RUN pip uninstall -y torchtext -RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook +RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook pillow>=9.1 # torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 # RUN pip install --no-cache -U torch torchvision From b122177aa16447a3f50e64b553abe0f1a76cf2d1 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 19:03:03 -0700 Subject: [PATCH 05/18] Update Dockerfile-cpu --- utils/docker/Dockerfile-cpu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/docker/Dockerfile-cpu b/utils/docker/Dockerfile-cpu index d30c07e81172..b169a16acd49 100644 --- a/utils/docker/Dockerfile-cpu +++ b/utils/docker/Dockerfile-cpu @@ -14,7 +14,7 @@ COPY requirements.txt . RUN python3 -m pip install --upgrade pip RUN pip install --no-cache -r requirements.txt albumentations gsutil notebook \ coremltools onnx onnx-simplifier onnxruntime openvino-dev tensorflow-cpu tensorflowjs \ - torch==1.11.0+cpu torchvision==0.12.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html + install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu # Create working directory RUN mkdir -p /usr/src/app From 754d759f16a2dd37c292664569cf31ffa12ef797 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 19:08:08 -0700 Subject: [PATCH 06/18] Update Dockerfile-cpu --- utils/docker/Dockerfile-cpu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/docker/Dockerfile-cpu b/utils/docker/Dockerfile-cpu index b169a16acd49..6cf99aaa9ee3 100644 --- a/utils/docker/Dockerfile-cpu +++ b/utils/docker/Dockerfile-cpu @@ -4,7 +4,7 @@ FROM ubuntu:latest # Install linux packages -RUN apt update +# RUN apt update RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt install -y tzdata RUN apt install -y python3-pip git zip curl htop screen libgl1-mesa-glx libglib2.0-0 RUN alias python=python3 From a8ba405a14d90691690bcfad74c9b12dd587f20b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 19:16:56 -0700 Subject: [PATCH 07/18] Update Dockerfile-cpu --- utils/docker/Dockerfile-cpu | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/utils/docker/Dockerfile-cpu b/utils/docker/Dockerfile-cpu index 6cf99aaa9ee3..5efef7853c56 100644 --- a/utils/docker/Dockerfile-cpu +++ b/utils/docker/Dockerfile-cpu @@ -4,9 +4,9 @@ FROM ubuntu:latest # Install linux packages -# RUN apt update -RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt install -y tzdata -RUN apt install -y python3-pip git zip curl htop screen libgl1-mesa-glx libglib2.0-0 +RUN apt-get update +RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt-get install -y tzdata +RUN apt-get install -y python3-pip git zip curl htop screen libgl1-mesa-glx libglib2.0-0 RUN alias python=python3 # Install python dependencies From 494bcfb396688ece052a75dcc5f07c8330dfed4e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 19:22:44 -0700 Subject: [PATCH 08/18] Update Dockerfile-cpu --- utils/docker/Dockerfile-cpu | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/utils/docker/Dockerfile-cpu b/utils/docker/Dockerfile-cpu index 5efef7853c56..05f5bc6919a6 100644 --- a/utils/docker/Dockerfile-cpu +++ b/utils/docker/Dockerfile-cpu @@ -4,9 +4,9 @@ FROM ubuntu:latest # Install linux packages -RUN apt-get update -RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt-get install -y tzdata -RUN apt-get install -y python3-pip git zip curl htop screen libgl1-mesa-glx libglib2.0-0 +RUN apt update +RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt install -y tzdata +RUN apt install -y python3-pip git zip curl htop screen libgl1-mesa-glx libglib2.0-0 RUN alias python=python3 # Install python dependencies @@ -31,7 +31,7 @@ ADD https://ultralytics.com/assets/Arial.ttf https://ultralytics.com/assets/Aria # Usage Examples ------------------------------------------------------------------------------------------------------- # Build and Push -# t=ultralytics/yolov5:latest-cpu && sudo docker build -f utils/docker/Dockerfile-cpu -t $t . && sudo docker push $t +# t=ultralytics/yolov5:latest-cpu-test && sudo docker build -f utils/docker/Dockerfile-cpu -t $t . && sudo docker push $t # Pull and Run # t=ultralytics/yolov5:latest-cpu && sudo docker pull $t && sudo docker run -it --ipc=host -v "$(pwd)"/datasets:/usr/src/datasets $t From 84f95f26f1d94ad59012ef3dd7017ed26aa42afd Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 19:22:53 -0700 Subject: [PATCH 09/18] Update Dockerfile-cpu --- utils/docker/Dockerfile-cpu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/docker/Dockerfile-cpu b/utils/docker/Dockerfile-cpu index 05f5bc6919a6..b169a16acd49 100644 --- a/utils/docker/Dockerfile-cpu +++ b/utils/docker/Dockerfile-cpu @@ -31,7 +31,7 @@ ADD https://ultralytics.com/assets/Arial.ttf https://ultralytics.com/assets/Aria # Usage Examples ------------------------------------------------------------------------------------------------------- # Build and Push -# t=ultralytics/yolov5:latest-cpu-test && sudo docker build -f utils/docker/Dockerfile-cpu -t $t . && sudo docker push $t +# t=ultralytics/yolov5:latest-cpu && sudo docker build -f utils/docker/Dockerfile-cpu -t $t . && sudo docker push $t # Pull and Run # t=ultralytics/yolov5:latest-cpu && sudo docker pull $t && sudo docker run -it --ipc=host -v "$(pwd)"/datasets:/usr/src/datasets $t From 0f3c37b5010d0e33a25bbf5af9bd5a4f7ae88c27 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 19:23:13 -0700 Subject: [PATCH 10/18] Update Dockerfile-cpu --- utils/docker/Dockerfile-cpu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/docker/Dockerfile-cpu b/utils/docker/Dockerfile-cpu index b169a16acd49..464bd251872f 100644 --- a/utils/docker/Dockerfile-cpu +++ b/utils/docker/Dockerfile-cpu @@ -14,7 +14,7 @@ COPY requirements.txt . RUN python3 -m pip install --upgrade pip RUN pip install --no-cache -r requirements.txt albumentations gsutil notebook \ coremltools onnx onnx-simplifier onnxruntime openvino-dev tensorflow-cpu tensorflowjs \ - install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu + torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu # Create working directory RUN mkdir -p /usr/src/app From 03b373a1ba685e42587683c8a96293023bb2a1da Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 19:30:08 -0700 Subject: [PATCH 11/18] Update Dockerfile-cpu --- utils/docker/Dockerfile | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/utils/docker/Dockerfile b/utils/docker/Dockerfile index 53fa43105563..6d1950b39b89 100644 --- a/utils/docker/Dockerfile +++ b/utils/docker/Dockerfile @@ -10,7 +10,7 @@ RUN apt update && apt install -y zip htop screen libgl1-mesa-glx COPY requirements.txt . RUN python -m pip install --upgrade pip RUN pip uninstall -y torchtext -RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook pillow>=9.1 +RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook Pillow>=9.1.0 # torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 # RUN pip install --no-cache -U torch torchvision @@ -32,10 +32,10 @@ ENV OMP_NUM_THREADS=8 # Usage Examples ------------------------------------------------------------------------------------------------------- # Build and Push -# t=ultralytics/yolov5:latest_2204_torchinstall && sudo docker build -f utils/docker/Dockerfile -t $t . && sudo docker push $t +# t=ultralytics/yolov5:latest-notorchinstall && sudo docker build -f utils/docker/Dockerfile -t $t . && sudo docker push $t # Pull and Run -# t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all $t +# t=ultralytics/yolov5:latest-nopytorchinstall && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all $t # Pull and Run with local directory access # t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all -v "$(pwd)"/datasets:/usr/src/datasets $t From a77bd0ac78d8e60c872ea1785a18673f140800a7 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 19:48:40 -0700 Subject: [PATCH 12/18] Update Dockerfile-cpu --- requirements.txt | 2 +- utils/docker/Dockerfile | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/requirements.txt b/requirements.txt index c04c43a97805..d12a13611d24 100755 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ opencv-python>=4.1.1 Pillow>=7.1.2 PyYAML>=5.3.1 requests>=2.23.0 -scipy>=1.5 +scipy>=1.4.1 torch>=1.7.0 torchvision>=0.8.1 tqdm>=4.41.0 diff --git a/utils/docker/Dockerfile b/utils/docker/Dockerfile index 6d1950b39b89..b36ecfc0a8ef 100644 --- a/utils/docker/Dockerfile +++ b/utils/docker/Dockerfile @@ -9,10 +9,10 @@ RUN apt update && apt install -y zip htop screen libgl1-mesa-glx # Install python dependencies COPY requirements.txt . RUN python -m pip install --upgrade pip -RUN pip uninstall -y torchtext -RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook Pillow>=9.1.0 +RUN pip uninstall -y Pillow torchtext +RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook # torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 -# RUN pip install --no-cache -U torch torchvision +RUN pip install --no-cache -U Pillow # Create working directory RUN mkdir -p /usr/src/app From f4f732926eefe41c9d64c68d388c93b24eb372b7 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 20:18:29 -0700 Subject: [PATCH 13/18] Update Dockerfile-cpu --- utils/docker/Dockerfile | 2 +- utils/docker/Dockerfile-cpu | 14 +++++++++----- 2 files changed, 10 insertions(+), 6 deletions(-) diff --git a/utils/docker/Dockerfile b/utils/docker/Dockerfile index b36ecfc0a8ef..ae2045617773 100644 --- a/utils/docker/Dockerfile +++ b/utils/docker/Dockerfile @@ -6,7 +6,7 @@ FROM nvcr.io/nvidia/pytorch:22.04-py3 # Install linux packages RUN apt update && apt install -y zip htop screen libgl1-mesa-glx -# Install python dependencies +# Install pip packages COPY requirements.txt . RUN python -m pip install --upgrade pip RUN pip uninstall -y Pillow torchtext diff --git a/utils/docker/Dockerfile-cpu b/utils/docker/Dockerfile-cpu index 464bd251872f..1fe4b1144942 100644 --- a/utils/docker/Dockerfile-cpu +++ b/utils/docker/Dockerfile-cpu @@ -6,13 +6,17 @@ FROM ubuntu:latest # Install linux packages RUN apt update RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt install -y tzdata -RUN apt install -y python3-pip git zip curl htop screen libgl1-mesa-glx libglib2.0-0 -RUN alias python=python3 +RUN apt install -y python3-pip git zip curl htop screen libgl1-mesa-glx libglib2.0-0 software-properties-common -# Install python dependencies +# Install python3.9 +RUN add-apt-repository ppa:deadsnakes/ppa -y +RUN apt install python3.9 python3.9-distutils -y +# RUN alias python=python3.9 + +# Install pip packages COPY requirements.txt . -RUN python3 -m pip install --upgrade pip -RUN pip install --no-cache -r requirements.txt albumentations gsutil notebook \ +RUN python3.9 -m pip install --upgrade pip +RUN python3.9 -m pip install --no-cache -r requirements.txt albumentations gsutil notebook \ coremltools onnx onnx-simplifier onnxruntime openvino-dev tensorflow-cpu tensorflowjs \ torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu From 915902fd083e446a50cf756db3b6aa1c4fd9536a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 20:20:23 -0700 Subject: [PATCH 14/18] Update Dockerfile --- utils/docker/Dockerfile | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/utils/docker/Dockerfile b/utils/docker/Dockerfile index ae2045617773..1687d7b29c79 100644 --- a/utils/docker/Dockerfile +++ b/utils/docker/Dockerfile @@ -11,7 +11,6 @@ COPY requirements.txt . RUN python -m pip install --upgrade pip RUN pip uninstall -y Pillow torchtext RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook - # torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 RUN pip install --no-cache -U Pillow # Create working directory @@ -32,10 +31,10 @@ ENV OMP_NUM_THREADS=8 # Usage Examples ------------------------------------------------------------------------------------------------------- # Build and Push -# t=ultralytics/yolov5:latest-notorchinstall && sudo docker build -f utils/docker/Dockerfile -t $t . && sudo docker push $t +# t=ultralytics/yolov5:latest && sudo docker build -f utils/docker/Dockerfile -t $t . && sudo docker push $t # Pull and Run -# t=ultralytics/yolov5:latest-nopytorchinstall && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all $t +# t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all $t # Pull and Run with local directory access # t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all -v "$(pwd)"/datasets:/usr/src/datasets $t From 9169534d179411f49dc0a9e0d3c700947f37f6ea Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 20:22:23 -0700 Subject: [PATCH 15/18] fix emoji --- utils/general.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/general.py b/utils/general.py index 7c89d02f2d9c..a8bc0cc1734c 100755 --- a/utils/general.py +++ b/utils/general.py @@ -467,7 +467,7 @@ def check_dataset(data, autodownload=True): if val: val = [Path(x).resolve() for x in (val if isinstance(val, list) else [val])] # val path if not all(x.exists() for x in val): - LOGGER.info(emojis('\nDataset not found ⚠️, missing paths %s' % [str(x) for x in val if not x.exists()])) + LOGGER.info(emojis('\nDataset not found ⚠, missing paths %s' % [str(x) for x in val if not x.exists()])) if s and autodownload: # download script t = time.time() root = path.parent if 'path' in data else '..' # unzip directory i.e. '../' From 82e9a2dd70f397cb92dda1c6f3be0318b003a81b Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 20:24:50 -0700 Subject: [PATCH 16/18] reduce scipy --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index d12a13611d24..4cf22b7093b4 100755 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ opencv-python>=4.1.1 Pillow>=7.1.2 PyYAML>=5.3.1 requests>=2.23.0 -scipy>=1.4.1 +scipy>=1.4.1 # Google Colab version torch>=1.7.0 torchvision>=0.8.1 tqdm>=4.41.0 From 79ecc569ddaa2a7af6b6406f01b454a507969dfa Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 20:35:41 -0700 Subject: [PATCH 17/18] add libpython3.9 --- utils/docker/Dockerfile-cpu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/docker/Dockerfile-cpu b/utils/docker/Dockerfile-cpu index 1fe4b1144942..b3468e1d0f07 100644 --- a/utils/docker/Dockerfile-cpu +++ b/utils/docker/Dockerfile-cpu @@ -10,7 +10,7 @@ RUN apt install -y python3-pip git zip curl htop screen libgl1-mesa-glx libglib2 # Install python3.9 RUN add-apt-repository ppa:deadsnakes/ppa -y -RUN apt install python3.9 python3.9-distutils -y +RUN apt install python3.9 python3.9-distutils libpython3.9 -y # RUN alias python=python3.9 # Install pip packages From fbaa1560b1b13da9e0adece82e95f753be1b408c Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 May 2022 20:50:43 -0700 Subject: [PATCH 18/18] Update Dockerfile --- utils/docker/Dockerfile | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/utils/docker/Dockerfile b/utils/docker/Dockerfile index 1687d7b29c79..f7554c7b48e4 100644 --- a/utils/docker/Dockerfile +++ b/utils/docker/Dockerfile @@ -1,7 +1,7 @@ # YOLOv5 🚀 by Ultralytics, GPL-3.0 license # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch -FROM nvcr.io/nvidia/pytorch:22.04-py3 +FROM nvcr.io/nvidia/pytorch:21.10-py3 # Install linux packages RUN apt update && apt install -y zip htop screen libgl1-mesa-glx @@ -9,9 +9,9 @@ RUN apt update && apt install -y zip htop screen libgl1-mesa-glx # Install pip packages COPY requirements.txt . RUN python -m pip install --upgrade pip -RUN pip uninstall -y Pillow torchtext -RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook -RUN pip install --no-cache -U Pillow +RUN pip uninstall -y torch torchvision torchtext +RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook \ + torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html # Create working directory RUN mkdir -p /usr/src/app