Add CUDA 11.8 (#9)

e23f4582 · Woosuk Kwon · GitHub · 5a3e6ebf · e23f4582
Unverified Commit e23f4582 authored Jul 29, 2024 by Woosuk Kwon Committed by GitHub Jul 29, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

.github/workflows/publish.yml .github/workflows/publish.yml +1 -1

No files found.
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -50,7 +50,7 @@ jobs:
          os: ['ubuntu-20.04']
          python-version: ['3.8', '3.9', '3.10', '3.11']
          pytorch-version: ['2.4.0']  # Should be synced with setup.py.
-          cuda-version: ['12.1']
+          cuda-version: ['12.1', '11.8']

    steps:
      - name: Checkout