diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml index d4c9852..e0b768a 100644 --- a/.github/workflows/publish.yml +++ b/.github/workflows/publish.yml @@ -51,7 +51,7 @@ jobs: os: ['ubuntu-20.04'] python-version: ['3.8', '3.9', '3.10', '3.11', '3.12'] pytorch-version: ['2.4.0'] # Must be the most recent version that meets requirements-cuda.txt. - cuda-version: ['11.8', '12.1'] + cuda-version: ['12.1'] steps: - name: Checkout