From b9ef2f14920df8e922ec3fc285e4b129f837cc4d Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Wed, 10 Jul 2024 13:24:23 -0400 Subject: [PATCH] Update validate_binaries.sh for windows --- .github/scripts/validate_binaries.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh index 0eab2d5e8..ab55870f7 100755 --- a/.github/scripts/validate_binaries.sh +++ b/.github/scripts/validate_binaries.sh @@ -71,7 +71,7 @@ else # Use case CUDA_VISIBLE_DEVICES: https://github.com/pytorch/pytorch/issues/128819 if [[ ${MATRIX_GPU_ARCH_TYPE} == 'cuda' ]]; then - python3 -c "import torch;import os;print(torch.cuda.device_count(), torch.__version__);os.environ['CUDA_VISIBLE_DEVICES']='0';print(torch.empty(2, device='cuda'))" + python -c "import torch;import os;print(torch.cuda.device_count(), torch.__version__);os.environ['CUDA_VISIBLE_DEVICES']='0';print(torch.empty(2, device='cuda'))" fi # this is optional step