Isolate 2.0.0 & cuda12

.github/workflows/cuda/cu120-Linux-env.sh

+#!/bin/bash
+
+CUDA_HOME=/usr/local/cuda-12.0
+LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
+PATH=${CUDA_HOME}/bin:${PATH}
+
+export FORCE_CUDA=1
+export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
+export CUDA_HOME=/usr/local/cuda-12.0
\ No newline at end of file

.github/workflows/cuda/cu120-Linux.sh

+#!/bin/bash
+
+# Strip the periods from the version number
+OS_VERSION=$(echo $(lsb_release -sr) | tr -d .)
+OS=ubuntu${OS_VERSION}
+
+wget -nv https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/cuda-${OS}.pin
+sudo mv cuda-${OS}.pin /etc/apt/preferences.d/cuda-repository-pin-600
+wget -nv https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda-repo-${OS}-12-0-local_12.0.0-525.60.13-1_amd64.deb
+
+sudo dpkg -i cuda-repo-${OS}-12-0-local_12.0.0-525.60.13-1_amd64.deb
+sudo cp /var/cuda-repo-${OS}-12-0-local/cuda-*-keyring.gpg /usr/share/keyrings/
+
+sudo apt-get -qq update
+sudo apt install cuda cuda-nvcc-12-0 cuda-libraries-dev-12-0
+sudo apt clean
+
+rm -f https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda-repo-${OS}-12-0-local_12.0.0-525.60.13-1_amd64.deb

.github/workflows/publish.yml

@@ -40,9 +40,12 @@ jobs:
       fail-fast: false
       matrix:
           os: [ubuntu-20.04, ubuntu-22.04]
-          python-version: ['3.7', '3.8', '3.9', '3.10']
-          torch-version: ['1.11.0', '1.12.0', '1.13.0']
-          cuda-version: ['113', '116', '117']
+          #python-version: ['3.7', '3.8', '3.9', '3.10']
+          #torch-version: ['1.11.0', '1.12.0', '1.13.0', '2.0.1']
+          #cuda-version: ['113', '116', '117', '120']
+          python-version: ['3.10']
+          torch-version: ['2.0.1']
+          cuda-version: ['120']
           exclude:
             # Nvidia only supports 11.7+ for ubuntu-22.04
             - os: ubuntu-22.04
@@ -57,9 +60,18 @@ jobs:
             # Torch only builds cuda 116 for 1.12.0+
             - cuda-version: '116'
               torch-version: '1.11.0'
+            # Torch only builds cuda 120 for 2.0.1+
+            - cuda-version: '120'
+              torch-version: '1.11.0'
+            - cuda-version: '120'
+              torch-version: '1.12.0'
+            - cuda-version: '120'
+              torch-version: '1.13.0'
             # 1.13.0 drops support for cuda 11.3
             - cuda-version: '113'
               torch-version: '1.13.0'
+            - cuda-version: '113'
+              torch-version: '2.0.1'
             # Fails with "Validation Error" on artifact upload
             - cuda-version: '117'
               torch-version: '1.13.0'