diff --git a/.github/workflows/cuda/cu116-Linux.sh b/.github/workflows/cuda/cu116-Linux.sh index 68e9ed4..f6ebbe3 100644 --- a/.github/workflows/cuda/cu116-Linux.sh +++ b/.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 diff --git a/.github/workflows/cuda/cu117-Linux-env.sh b/.github/workflows/cuda/cu117-Linux-env.sh new file mode 100644 index 0000000..ab432d1 --- /dev/null +++ b/.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 \ No newline at end of file diff --git a/.github/workflows/cuda/cu117-Linux.sh b/.github/workflows/cuda/cu117-Linux.sh new file mode 100644 index 0000000..40e66f3 --- /dev/null +++ b/.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 diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml index 0fc0281..2475ba0 100644 --- a/.github/workflows/publish.yml +++ b/.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