From 506d259fd6ca12fc969592857fc8c811444494a4 Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Thu, 31 Aug 2023 17:21:27 +0200 Subject: [PATCH] install third party dependencies with ltt as well --- .github/scripts/setup-env.sh | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/.github/scripts/setup-env.sh b/.github/scripts/setup-env.sh index 4b852efd9b7..f9b9b71f805 100755 --- a/.github/scripts/setup-env.sh +++ b/.github/scripts/setup-env.sh @@ -95,7 +95,10 @@ python setup.py egg_info # The requires.txt cannot be used with `pip install -r` directly. The requirements are listed at the top and the # optional dependencies come in non-standard syntax after a blank line. Thus, we just extract the header. sed -e '/^$/,$d' *.egg-info/requires.txt | tee requirements.txt -pip install --progress-bar=off -r requirements.txt +ltt install --progress-bar=off \ + --pytorch-computation-backend="${GPU_ARCH_TYPE}${GPU_ARCH_VERSION}" \ + --pytorch-channel="${CHANNEL}" \ + -r requirements.txt echo '::endgroup::' echo '::group::Install TorchVision'