Skip to content

Commit 212b91e

Browse files
committed
update tested CUDA versions in CI
1 parent 2eeed03 commit 212b91e

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

.github/workflows/install.yml

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -71,22 +71,24 @@ jobs:
7171
- windows-latest
7272
- macos-latest
7373
python-version:
74-
- "3.8"
7574
- "3.9"
7675
- "3.10"
7776
- "3.11"
7877
pytorch-computation-backend:
7978
- cpu
80-
- cu117
8179
- cu118
80+
- cu121
81+
- cu124
8282
exclude:
83-
- os: macos-latest
84-
pytorch-computation-backend: cu117
8583
- os: macos-latest
8684
pytorch-computation-backend: cu118
85+
- os: macos-latest
86+
pytorch-computation-backend: cu121
87+
- os: macos-latest
88+
pytorch-computation-backend: cu124
8789
# TODO: find a way to test this
8890
# - os: ubuntu-latest
89-
# pytorch-computation-backend: rocm5.4.2
91+
# pytorch-computation-backend: rocm6.2
9092
fail-fast: false
9193

9294
runs-on: ${{ matrix.os }}
@@ -108,7 +110,7 @@ jobs:
108110
- name: Install torch
109111
run:
110112
ltt install --pytorch-computation-backend=${{
111-
matrix.pytorch-computation-backend }} torch==2.0.0
113+
matrix.pytorch-computation-backend }} torch==2.5.1
112114

113115
- name: Check computation backend
114116
shell: python

0 commit comments

Comments
 (0)