Add CUDA 11.2 support to manywheels scripts (#617)

Author: janeyx99

.gitignore

@@ -78,3 +78,7 @@ node_modules
 
 # mypy
 **/.mypy_cache/
+
+# Mac directory meta
+.DS_Store
+*/**/.DS_Store

manywheel/build.sh

@@ -52,6 +52,10 @@ cuda_version_nodot=$(echo $CUDA_VERSION | tr -d '.')
 
 TORCH_CUDA_ARCH_LIST="3.7;5.0;6.0;7.0"
 case ${CUDA_VERSION} in
+    11.2)
+        TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};7.5;8.0;8.6"
+        EXTRA_CAFFE2_CMAKE_FLAGS+=("-DATEN_NO_TEST=ON")
+        ;;
     11.1)
         TORCH_CUDA_ARCH_LIST="5.0;7.0;8.0;8.6"  # removing some to prevent bloated binary size
         EXTRA_CAFFE2_CMAKE_FLAGS+=("-DATEN_NO_TEST=ON")
@@ -207,7 +211,7 @@ DEPS_SONAME=(
 )
 elif [[ $CUDA_VERSION == "11.1" ]]; then
 DEPS_LIST=(
-    "/usr/local/cuda/lib64/libcudart.so.11.0"   # CUDA 11.1 usues libcudart11.0 for backwards compat
+    "/usr/local/cuda/lib64/libcudart.so.11.0"   # CUDA 11.1 uses libcudart11.0 for backwards compat
     "/usr/local/cuda/lib64/libnvToolsExt.so.1"
     "/usr/local/cuda/lib64/libnvrtc.so.11.1"
     "/usr/local/cuda/lib64/libnvrtc-builtins.so"
@@ -221,6 +225,22 @@ DEPS_SONAME=(
     "libnvrtc-builtins.so"
     "libgomp.so.1"
 )
+elif [[ $CUDA_VERSION == "11.2" ]]; then
+DEPS_LIST=(
+    "/usr/local/cuda/lib64/libcudart.so.11.2.72"
+    "/usr/local/cuda/lib64/libnvToolsExt.so.1"
+    "/usr/local/cuda/lib64/libnvrtc.so.11.2"
+    "/usr/local/cuda/lib64/libnvrtc-builtins.so"
+    "$LIBGOMP_PATH"
+)
+
+DEPS_SONAME=(
+    "libcudart.so.11.2.72"
+    "libnvToolsExt.so.1"
+    "libnvrtc.so.11.2"
+    "libnvrtc-builtins.so"
+    "libgomp.so.1"
+)
 else
     echo "Unknown cuda version $CUDA_VERSION"
     exit 1

manywheel/deploy.sh

@@ -16,7 +16,7 @@ for rocm_version in 3.7 3.8 3.9 3.10 4.0; do
     )
 done
 
-for cuda_version in 9.2 10.1 10.2 11.0 11.1; do
+for cuda_version in 9.2 10.1 10.2 11.0 11.1 11.2; do
     (
         set -x
         DOCKER_BUILDKIT=1 docker build \