From 565615c603bc83ff0215cf62bc4d907b27041215 Mon Sep 17 00:00:00 2001
From: Pierce Freeman <piercefreeman@gmail.com>
Date: Thu, 10 Aug 2023 19:54:29 -0700
Subject: [PATCH] Isolate 2.0.0 & cuda12

---
 .github/workflows/cuda/cu120-Linux-env.sh |  9 +++++++++
 .github/workflows/cuda/cu120-Linux.sh     | 18 ++++++++++++++++++
 .github/workflows/publish.yml             | 18 +++++++++++++++---
 3 files changed, 42 insertions(+), 3 deletions(-)
 create mode 100644 .github/workflows/cuda/cu120-Linux-env.sh
 create mode 100644 .github/workflows/cuda/cu120-Linux.sh

diff --git a/.github/workflows/cuda/cu120-Linux-env.sh b/.github/workflows/cuda/cu120-Linux-env.sh
new file mode 100644
index 0000000..37917cc
--- /dev/null
+++ b/.github/workflows/cuda/cu120-Linux-env.sh
@@ -0,0 +1,9 @@
+#!/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
diff --git a/.github/workflows/cuda/cu120-Linux.sh b/.github/workflows/cuda/cu120-Linux.sh
new file mode 100644
index 0000000..56996de
--- /dev/null
+++ b/.github/workflows/cuda/cu120-Linux.sh
@@ -0,0 +1,18 @@
+#!/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
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index 83c4b48..05eaaad 100644
--- a/.github/workflows/publish.yml
+++ b/.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'