Add CUDA 11.7

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

@@ -7,6 +7,7 @@ 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/11.6.2/local_installers/cuda-repo-${OS}-11-6-local_11.6.2-510.47.03-1_amd64.deb
+
 sudo dpkg -i cuda-repo-${OS}-11-6-local_11.6.2-510.47.03-1_amd64.deb
 sudo apt-key add /var/cuda-repo-${OS}-11-6-local/7fa2af80.pub
 

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

@@ -0,0 +1,9 @@
+#!/bin/bash
+
+CUDA_HOME=/usr/local/cuda-11.7
+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-11.7

.github/workflows/cuda/cu117-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/11.7.0/local_installers/cuda-repo-${OS}-11-7-local_11.7.0-515.43.04-1_amd64.deb
+
+sudo dpkg -i cuda-repo-${OS}-11-7-local_11.7.0-515.43.04-1_amd64.deb
+sudo cp /var/cuda-repo-${OS}-11-7-local/cuda-*-keyring.gpg /usr/share/keyrings/
+
+sudo apt-get -qq update
+sudo apt install cuda cuda-nvcc-11-7 cuda-libraries-dev-11-7
+sudo apt clean
+
+rm -f https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda-repo-${OS}-11-7-local_11.7.0-515.43.04-1_amd64.deb

.github/workflows/publish.yml

@@ -40,12 +40,15 @@ jobs:
           #python-version: ['3.7', '3.8', '3.9', '3.10']
           python-version: ['3.10']
           torch-version: [1.11.0]
-          cuda-version: ['113']
+          cuda-version: ['113', '117']
           #torch-version: [1.11.0, 1.12.0, 1.12.1]
           #cuda-version: ['113', '116']
-          #exclude:
-          #  - torch-version: 1.11.0
-          #    cuda-version: '116'
+          exclude:
+            # Nvidia only supports 11.7+ for ubuntu-22.04
+            - os: ubuntu-22.04
+              cuda-version: '116'
+            - os: ubuntu-22.04
+              cuda-version: '113'
 
     steps:
       - name: Checkout