From 3b852c4a0abf9f163c115b6521ce19b8615f3a7a Mon Sep 17 00:00:00 2001 From: vfdev Date: Thu, 8 Dec 2022 14:00:31 +0100 Subject: [PATCH 1/3] Update test-linux-gpu.yml --- .github/workflows/test-linux-gpu.yml | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test-linux-gpu.yml b/.github/workflows/test-linux-gpu.yml index a4d938f23ed..396f2a3c49e 100644 --- a/.github/workflows/test-linux-gpu.yml +++ b/.github/workflows/test-linux-gpu.yml @@ -21,7 +21,7 @@ jobs: fail-fast: false uses: pytorch/test-infra/.github/workflows/linux_job.yml@main with: - runner: linux.g5.4xlarge.nvidia.gpu + runner: linux.4xlarge.nvidia.gpu repository: pytorch/vision gpu-arch-type: cuda gpu-arch-version: ${{ matrix.cuda_arch_version }} @@ -51,8 +51,11 @@ jobs: conda install \ --yes \ -c "pytorch-${CHANNEL}" \ - -c nvidia "pytorch-${CHANNEL}"::pytorch[build="*${VERSION}*"] \ + -c nvidia pytorch \ "${CUDATOOLKIT}" + + python3 -c "import torch; exit(not torch.cuda.is_available())" + python3 setup.py develop python3 -m pip install pytest pytest-mock 'av<10' From 742c202246e94f3904aa87e99da3b91651a09115 Mon Sep 17 00:00:00 2001 From: vfdev Date: Thu, 8 Dec 2022 14:11:12 +0100 Subject: [PATCH 2/3] Update .github/workflows/test-linux-gpu.yml Co-authored-by: Philip Meier --- .github/workflows/test-linux-gpu.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/test-linux-gpu.yml b/.github/workflows/test-linux-gpu.yml index 396f2a3c49e..3c7271dc6f8 100644 --- a/.github/workflows/test-linux-gpu.yml +++ b/.github/workflows/test-linux-gpu.yml @@ -51,7 +51,8 @@ jobs: conda install \ --yes \ -c "pytorch-${CHANNEL}" \ - -c nvidia pytorch \ + -c nvidia \ + pytorch \ "${CUDATOOLKIT}" python3 -c "import torch; exit(not torch.cuda.is_available())" From dd2d6da1a83920a1203aaee4e8e9185c9ec634a0 Mon Sep 17 00:00:00 2001 From: vfdev Date: Thu, 8 Dec 2022 16:36:47 +0100 Subject: [PATCH 3/3] Update test-linux-gpu.yml --- .github/workflows/test-linux-gpu.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test-linux-gpu.yml b/.github/workflows/test-linux-gpu.yml index 3c7271dc6f8..37a4256a19d 100644 --- a/.github/workflows/test-linux-gpu.yml +++ b/.github/workflows/test-linux-gpu.yml @@ -21,7 +21,7 @@ jobs: fail-fast: false uses: pytorch/test-infra/.github/workflows/linux_job.yml@main with: - runner: linux.4xlarge.nvidia.gpu + runner: linux.g5.4xlarge.nvidia.gpu repository: pytorch/vision gpu-arch-type: cuda gpu-arch-version: ${{ matrix.cuda_arch_version }}