We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 356d1a4 commit de6b95cCopy full SHA for de6b95c
.github/workflows/regression_test.yml
@@ -23,7 +23,7 @@ jobs:
23
include:
24
- name: CUDA 2.2.2
25
runs-on: linux.g5.12xlarge.nvidia.gpu
26
- torch-spec: 'torch==2.2.2 numpy==1.26'
+ torch-spec: 'torch==2.2.2 "numpy<2" '
27
gpu-arch-type: "cuda"
28
gpu-arch-version: "12.1"
29
- name: CUDA 2.3
@@ -38,7 +38,7 @@ jobs:
38
39
- name: CPU 2.2.2
40
runs-on: linux.4xlarge
41
- torch-spec: 'torch==2.2.2 --index-url https://download.pytorch.org/whl/cpu numpy==1.26'
+ torch-spec: 'torch==2.2.2 --index-url https://download.pytorch.org/whl/cpu "numpy<2" '
42
gpu-arch-type: "cpu"
43
gpu-arch-version: ""
44
- name: CPU 2.3
0 commit comments