From 894c99fd183b3547cf3cd3f7b17654273a1aa263 Mon Sep 17 00:00:00 2001 From: Roman Donchenko Date: Mon, 29 Apr 2024 20:04:25 +0300 Subject: [PATCH] Use CPU PyTorch for testing the SDK Not only is the GPU version of PyTorch much bigger than the CPU version, but it also pulls in CUDA, which is enormous. We don't (and can't) use any GPU features in our tests, so we don't need the GPU version. Eliminating it saves ~4GB of disk space, which is precious when it comes to the GitHub actions runners. --- .github/workflows/full.yml | 3 ++- .github/workflows/main.yml | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/.github/workflows/full.yml b/.github/workflows/full.yml index c3401a744fd..6d45c5e571b 100644 --- a/.github/workflows/full.yml +++ b/.github/workflows/full.yml @@ -160,7 +160,8 @@ jobs: - name: Install SDK run: | pip3 install -r ./tests/python/requirements.txt \ - -e './cvat-sdk[pytorch]' -e ./cvat-cli + -e './cvat-sdk[pytorch]' -e ./cvat-cli \ + --extra-index-url https://download.pytorch.org/whl/cpu - name: Running REST API and SDK tests id: run_tests diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index c2d7617f7b3..e1492144df3 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -170,7 +170,8 @@ jobs: - name: Install SDK run: | pip3 install -r ./tests/python/requirements.txt \ - -e './cvat-sdk[pytorch]' -e ./cvat-cli + -e './cvat-sdk[pytorch]' -e ./cvat-cli \ + --extra-index-url https://download.pytorch.org/whl/cpu - name: Run REST API and SDK tests id: run_tests