Add new wheels (#23)

* add torch 1.13.1

* add torch2.0.0

* fix typo and exclude some cases

* fix cuda install

* exclude some case

* fix typo

* remove torch 1.11.0

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

+#!/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"
\ No newline at end of file

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

+#!/bin/bash
+
+OS=ubuntu1804
+
+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.1/local_installers/cuda-repo-${OS}-11-7-local_11.7.1-515.65.01-1_amd64.deb
+sudo dpkg -i cuda-repo-${OS}-11-7-local_11.7.1-515.65.01-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.1/local_installers/cuda-repo-${OS}-11-7-local_11.7.1-515.65.01-1_amd64.deb
\ No newline at end of file

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

+#!/bin/bash
+
+CUDA_HOME=/usr/local/cuda-11.8
+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"
\ No newline at end of file

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

+#!/bin/bash
+
+OS=ubuntu1804
+
+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.8.0/local_installers/cuda-repo-${OS}-11-8-local_11.8.0-520.61.05-1_amd64.deb
+sudo dpkg -i cuda-repo-${OS}-11-8-local_11.8.0-520.61.05-1_amd64.deb
+sudo cp /var/cuda-repo-${OS}-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
+
+sudo apt-get -qq update
+sudo apt install cuda cuda-nvcc-11-8 cuda-libraries-dev-11-8
+sudo apt clean
+
+rm -f https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-${OS}-11-8-local_11.8.0-520.61.05-1_amd64.deb
\ No newline at end of file

.github/workflows/publish.yml

@@ -39,10 +39,26 @@ jobs:
           # os: [ubuntu-20.04]
           os: [ubuntu-18.04]
           python-version: ['3.7', '3.8', '3.9', '3.10']
-          torch-version: [1.11.0, 1.12.0, 1.12.1]
-          cuda-version: ['113', '116']
+          torch-version: [1.12.0, 1.12.1, 1.13.1, 2.0.0]
+          cuda-version: ['113', '116', '117', '118']
           exclude:
-            - torch-version: 1.11.0
+            - torch-version: 1.12.0
+              cuda-version: '117'
+            - torch-version: 1.12.0
+              cuda-version: '118'
+            - torch-version: 1.12.1
+              cuda-version: '117'
+            - torch-version: 1.12.1
+              cuda-version: '118'
+            - torch-version: 1.13.1
+              cuda-version: '113'
+            - torch-version: 1.13.1
+              cuda-version: '118'
+            - torch-version: 2.0.0
+              python-version: '3.7'
+            - torch-version: 2.0.0
+              cuda-version: '113'
+            - torch-version: 2.0.0
               cuda-version: '116'
 
     steps:

setup.py

@@ -72,9 +72,11 @@ print("\n\ntorch.__version__  = {}\n\n".format(torch.__version__))
 TORCH_MAJOR = int(torch.__version__.split('.')[0])
 TORCH_MINOR = int(torch.__version__.split('.')[1])
 
-if not (TORCH_MAJOR >= 1 and TORCH_MINOR >= 4):
-      raise RuntimeError("Requires Pytorch 1.4 or newer.\n" +
-                         "The latest stable release can be obtained from https://pytorch.org/")
+if not ( (TORCH_MAJOR >= 1 and TORCH_MINOR >= 4)
+        or (TORCH_MAJOR > 1)
+    ):
+    raise RuntimeError("Requires Pytorch 1.4 or newer.\n" +
+                        "The latest stable release can be obtained from https://pytorch.org/")
 
 cmdclass = {}
 ext_modules = []