Update validate_binaries.sh for windows

atalman · Jul 10, 2024 · b9ef2f1 · b9ef2f1
1 parent 987dcb0
commit b9ef2f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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