Support torch 1.12.0

scripts/github_actions/generate_build_matrix.py

@@ -112,9 +112,13 @@ def generate_build_matrix(enable_cuda, for_windows, test_only_latest_torch):
             if not for_windows
             else ["11.3.1", "11.5.2"],
         },
+        "1.12.0": {
+            "python-version": ["3.7", "3.8", "3.9", "3.10"],
+            "cuda": ["10.2", "11.3", "11.6"],
+        },
     }
     if test_only_latest_torch:
-        latest = "1.11.0"
+        latest = "1.12.0"
         matrix = {latest: matrix[latest]}
 
     ans = []

scripts/github_actions/install_cuda.sh

@@ -43,6 +43,9 @@ case "$cuda" in
   11.5)
     url=https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_495.29.05_linux.run
     ;;
+  11.6)
+    url=https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
+    ;;
   *)
     echo "Unknown cuda version: $cuda"
     exit 1

scripts/github_actions/install_cudnn.sh

@@ -36,9 +36,9 @@ case $cuda in
   11.5)
     filename=cudnn-11.3-linux-x64-v8.2.0.53.tgz
     ;;
-  # 11.5)
+  11.6)
-  #   filename=cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz
+    filename=cudnn-11.3-linux-x64-v8.2.0.53.tgz
-  #   ;;
+    ;;
   *)
     echo "Unsupported cuda version: $cuda"
     exit 1

scripts/github_actions/install_torch.sh

@@ -125,6 +125,23 @@ case ${torch} in
         ;;
     esac
     ;;
+  1.12.*)
+    case ${cuda} in
+      10.2)
+        package="torch==${torch}"
+        # Leave it empty to use PyPI.
+        url=
+        ;;
+      11.3)
+        package="torch==${torch}+cu113"
+        url=https://download.pytorch.org/whl/torch_stable.html
+        ;;
+      11.6)
+        package="torch==${torch}+cu116"
+        url=https://download.pytorch.org/whl/torch_stable.html
+        ;;
+    esac
+    ;;
   *)
     echo "Unsupported PyTorch version: ${torch}"
     exit 1