Add pip install wheel to avoid legacy setup.py install (ultralyti…

…cs#8597) * Update ci-testing with `pip install wheel` * Update ci-testing.yml * Update dockerfiles
UnglvKitDe · Jul 16, 2022 · 6e86af3 · 6e86af3
1 parent 5c45a4b
commit 6e86af3
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml
@@ -32,7 +32,7 @@ jobs:
       #    restore-keys: ${{ runner.os }}-Benchmarks-
       - name: Install requirements
         run: |
-          python -m pip install --upgrade pip
+          python -m pip install --upgrade pip wheel
           pip install -r requirements.txt coremltools openvino-dev tensorflow-cpu --extra-index-url https://download.pytorch.org/whl/cpu
           python --version
           pip --version
@@ -77,7 +77,7 @@ jobs:
           restore-keys: ${{ runner.os }}-${{ matrix.python-version }}-pip-
       - name: Install requirements
         run: |
-          python -m pip install --upgrade pip
+          python -m pip install --upgrade pip wheel
           pip install -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu
           python --version
           pip --version

diff --git a/utils/docker/Dockerfile b/utils/docker/Dockerfile
@@ -14,7 +14,7 @@ RUN apt update && apt install --no-install-recommends -y zip htop screen libgl1-
 
 # Install pip packages
 COPY requirements.txt .
-RUN python -m pip install --upgrade pip
+RUN python -m pip install --upgrade pip wheel
 RUN pip uninstall -y torch torchvision torchtext Pillow
 RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook Pillow>=9.1.0 \
     'opencv-python<4.6.0.66' \

diff --git a/utils/docker/Dockerfile-arm64 b/utils/docker/Dockerfile-arm64
@@ -17,7 +17,7 @@ RUN apt install --no-install-recommends -y python3-pip git zip curl htop gcc \
 
 # Install pip packages
 COPY requirements.txt .
-RUN python3 -m pip install --upgrade pip
+RUN python3 -m pip install --upgrade pip wheel
 RUN pip install --no-cache -r requirements.txt gsutil notebook \
     tensorflow-aarch64
     # tensorflowjs \

diff --git a/utils/docker/Dockerfile-cpu b/utils/docker/Dockerfile-cpu
@@ -16,7 +16,7 @@ RUN apt install --no-install-recommends -y python3-pip git zip curl htop libgl1-
 
 # Install pip packages
 COPY requirements.txt .
-RUN python3 -m pip install --upgrade pip
+RUN python3 -m pip install --upgrade pip wheel
 RUN pip install --no-cache -r requirements.txt albumentations gsutil notebook \
     coremltools onnx onnx-simplifier onnxruntime openvino-dev tensorflow-cpu tensorflowjs \
     --extra-index-url https://download.pytorch.org/whl/cpu