Skip to content

Commit da257b5

Browse files
install torchvision to resolve ci issue (#4203)
1 parent 3ad1067 commit da257b5

1 file changed

Lines changed: 8 additions & 8 deletions

File tree

.github/workflows/regression_test.yml

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)