diff --git a/.flexci/windows/test.ps1 b/.flexci/windows/test.ps1 index 418dcac4..8b183dab 100644 --- a/.flexci/windows/test.ps1 +++ b/.flexci/windows/test.ps1 @@ -15,7 +15,10 @@ if ($test -eq "torch110") { ActivatePython 3.8 RunOrDie python -m pip install -U pip "setuptools<59.6" RunOrDieWithRetry 3 python -m pip install torch==1.10.* torchvision==0.11.* -f https://download.pytorch.org/whl/cu113/torch_stable.html - RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x + # NOTE(linsho): + # Due to reports of the ONNX build for Windows being broken, we are temporarily limiting the version of ONNX. + # cf. https://github.com/onnx/onnx/issues/6267 + RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x "onnx<=1.16.1" } elseif ($test -eq "torch111") { # PyTorch 1.11 + Python 3.9 @@ -23,7 +26,10 @@ if ($test -eq "torch110") { ActivatePython 3.9 RunOrDie python -m pip install -U pip "setuptools<59.6" RunOrDieWithRetry 3 python -m pip install torch==1.11.* torchvision==0.12.* -f https://download.pytorch.org/whl/cu113/torch_stable.html - RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x + # NOTE(linsho): + # Due to reports of the ONNX build for Windows being broken, we are temporarily limiting the version of ONNX. + # cf. https://github.com/onnx/onnx/issues/6267 + RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x "onnx<=1.16.1" } elseif ($test -eq "torch112") { # PyTorch 1.12 + Python 3.10 @@ -31,7 +37,10 @@ if ($test -eq "torch110") { ActivatePython 3.10 RunOrDie python -m pip install -U pip "setuptools<59.6" RunOrDieWithRetry 3 python -m pip install torch==1.12.* torchvision==0.13.* -f https://download.pytorch.org/whl/cu113/torch_stable.html - RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x + # NOTE(linsho): + # Due to reports of the ONNX build for Windows being broken, we are temporarily limiting the version of ONNX. + # cf. https://github.com/onnx/onnx/issues/6267 + RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x "onnx<=1.16.1" } elseif ($test -eq "torch113") { # PyTorch 1.13 + Python 3.10 @@ -39,7 +48,10 @@ if ($test -eq "torch110") { ActivatePython 3.10 RunOrDie python -m pip install -U pip "setuptools<59.6" RunOrDieWithRetry 3 python -m pip install torch==1.13.* torchvision==0.14.* -f https://download.pytorch.org/whl/cu117/torch_stable.html - RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x + # NOTE(linsho): + # Due to reports of the ONNX build for Windows being broken, we are temporarily limiting the version of ONNX. + # cf. https://github.com/onnx/onnx/issues/6267 + RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x "onnx<=1.16.1" } elseif ($test -eq "torch200") { # PyTorch 2.0 + Python 3.10 @@ -47,7 +59,10 @@ if ($test -eq "torch110") { ActivatePython 3.10 RunOrDie python -m pip install -U pip "setuptools<59.6" RunOrDieWithRetry 3 python -m pip install torch==2.0.* torchvision==0.15.* -f https://download.pytorch.org/whl/cu117/torch_stable.html - RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x + # NOTE(linsho): + # Due to reports of the ONNX build for Windows being broken, we are temporarily limiting the version of ONNX. + # cf. https://github.com/onnx/onnx/issues/6267 + RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x "onnx<=1.16.1" } elseif ($test -eq "torch201") { # PyTorch 2.1 + Python 3.10 @@ -55,7 +70,10 @@ if ($test -eq "torch110") { ActivatePython 3.10 RunOrDie python -m pip install -U pip "setuptools<59.6" RunOrDieWithRetry 3 python -m pip install torch==2.1.* torchvision==0.16.* -f https://download.pytorch.org/whl/cu118/torch_stable.html - RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x + # NOTE(linsho): + # Due to reports of the ONNX build for Windows being broken, we are temporarily limiting the version of ONNX. + # cf. https://github.com/onnx/onnx/issues/6267 + RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda11x "onnx<=1.16.1" } elseif ($test -eq "torch202") { # PyTorch 2.2 + Python 3.10 @@ -63,7 +81,10 @@ if ($test -eq "torch110") { ActivatePython 3.10 RunOrDie python -m pip install -U pip "setuptools<59.6" RunOrDieWithRetry 3 python -m pip install torch==2.2.* torchvision==0.17.* -f https://download.pytorch.org/whl/cu121/torch_stable.html - RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda12x + # NOTE(linsho): + # Due to reports of the ONNX build for Windows being broken, we are temporarily limiting the version of ONNX. + # cf. https://github.com/onnx/onnx/issues/6267 + RunOrDie python -m pip install -r tests/requirements.txt cupy-cuda12x "onnx<=1.16.1" } else { throw "Unsupported test variant: $test"