[dev] enable cuda12.1 build (#6567)

cmake/modules/CUDA.cmake

@@ -10,11 +10,11 @@ endif()
 include(CheckCXXCompilerFlag)
 check_cxx_compiler_flag("-std=c++17"   SUPPORT_CXX17)
 
-set(dgl_known_gpu_archs "35" "50" "60" "70")
+set(dgl_known_gpu_archs "35" "50" "60" "70" "75")
 set(dgl_cuda_arch_ptx "70")
 if (CUDA_VERSION_MAJOR GREATER_EQUAL "11")
-  list(APPEND dgl_known_gpu_archs "80")
-  set(dgl_cuda_arch_ptx "80")
+  list(APPEND dgl_known_gpu_archs "80" "86")
+  set(dgl_cuda_arch_ptx "80" "86")
 endif()
 # CMake 3.5 doesn't support VERSION_GREATER_EQUAL
 if (NOT CUDA_VERSION VERSION_LESS "11.8")

script/create_dev_conda_env.sh

 #!/bin/bash
 
-readonly CUDA_VERSIONS="11.6,11.7,11.8"
+readonly CUDA_VERSIONS="11.6,11.7,11.8,12.1"
 readonly TORCH_VERSION="1.13.0"
 readonly PYTHON_VERSION="3.8"