diff --git a/.github/workflows/cuda/install.sh b/.github/workflows/cuda/install.sh index d33c08f..73824ac 100755 --- a/.github/workflows/cuda/install.sh +++ b/.github/workflows/cuda/install.sh @@ -13,6 +13,9 @@ case "${cuda_version}" in "cu118") url=https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61.05-1_amd64.deb ;; + "cu121") + url=https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda-repo-ubuntu2004-12-1-local_12.1.1-530.30.02-1_amd64.deb + ;; *) >&2 echo "Unsupported cuda_version: ${cuda_version}" exit 1 diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml index 9d50280..7871cba 100644 --- a/.github/workflows/release.yml +++ b/.github/workflows/release.yml @@ -20,11 +20,17 @@ jobs: matrix: os: [ubuntu-20.04] python-version: ['3.10', '3.11'] - torch-version: ['1.13.1', '2.0.1'] - cuda-version: ['cpu', 'cu117', 'cu118'] + torch-version: ['1.13.1', '2.0.1', '2.1.0'] + cuda-version: ['cpu', 'cu117', 'cu118', 'cu121'] exclude: - torch-version: 1.13.1 cuda-version: cu118 + - torch-version: 1.13.1 + cuda-version: cu121 + - torch-version: 2.0.1 + cuda-version: cu121 + - torch-version: 2.1.0 + cuda-version: cu117 steps: - name: Free disk space run: |