@@ -25,12 +25,12 @@ jobs:
2525 include :
2626 - name : CUDA Nightly
2727 runs-on : linux.g5.12xlarge.nvidia.gpu
28- torch-spec : ' --pre torch --index-url https://download.pytorch.org/whl/nightly/cu126'
28+ torch-spec : ' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu126'
2929 gpu-arch-type : " cuda"
3030 gpu-arch-version : " 12.6"
3131 - name : CPU Nightly
3232 runs-on : linux.4xlarge
33- torch-spec : ' --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu'
33+ torch-spec : ' --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu'
3434 gpu-arch-type : " cpu"
3535 gpu-arch-version : " "
3636
@@ -61,38 +61,38 @@ jobs:
6161 include :
6262 - name : CUDA 2.8
6363 runs-on : linux.g5.12xlarge.nvidia.gpu
64- torch-spec : ' torch==2.8.0'
64+ torch-spec : ' torch==2.8.0 torchvision==0.23.0 '
6565 gpu-arch-type : " cuda"
6666 gpu-arch-version : " 12.6"
6767 dev-requirements-overrides : " "
6868 - name : CUDA 2.9
6969 runs-on : linux.g5.12xlarge.nvidia.gpu
70- torch-spec : ' torch==2.9.1'
70+ torch-spec : ' torch==2.9.1 torchvision==0.24.1 '
7171 gpu-arch-type : " cuda"
7272 gpu-arch-version : " 12.6"
7373 dev-requirements-overrides : " "
7474 - name : CUDA 2.10
7575 runs-on : linux.g5.12xlarge.nvidia.gpu
76- torch-spec : ' torch==2.10.0'
76+ torch-spec : ' torch==2.10.0 torchvision==0.25.0 '
7777 gpu-arch-type : " cuda"
7878 gpu-arch-version : " 12.6"
7979 dev-requirements-overrides : " "
8080
8181 - name : CPU 2.8
8282 runs-on : linux.4xlarge
83- torch-spec : ' torch==2.8.0 --index-url https://download.pytorch.org/whl/cpu'
83+ torch-spec : ' torch==2.8.0 torchvision==0.23.0 --index-url https://download.pytorch.org/whl/cpu'
8484 gpu-arch-type : " cpu"
8585 gpu-arch-version : " "
8686 dev-requirements-overrides : " "
8787 - name : CPU 2.9
8888 runs-on : linux.4xlarge
89- torch-spec : ' torch==2.9.1 --index-url https://download.pytorch.org/whl/cpu'
89+ torch-spec : ' torch==2.9.1 torchvision==0.24.1 --index-url https://download.pytorch.org/whl/cpu'
9090 gpu-arch-type : " cpu"
9191 gpu-arch-version : " "
9292 dev-requirements-overrides : " "
9393 - name : CPU 2.10
9494 runs-on : linux.4xlarge
95- torch-spec : ' torch==2.10.0 --index-url https://download.pytorch.org/whl/cpu'
95+ torch-spec : ' torch==2.10.0 torchvision==0.25.0 --index-url https://download.pytorch.org/whl/cpu'
9696 gpu-arch-type : " cpu"
9797 gpu-arch-version : " "
9898 dev-requirements-overrides : " "
0 commit comments