Skip to content

GitLab

Unverified Commit 19096c6a authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

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

parent 2f47c241
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
cmake/modules/CUDA.cmake
View file @ 19096c6a
...@@ -10,11 +10,11 @@ endif() ...@@ -10,11 +10,11 @@ endif()
include(CheckCXXCompilerFlag) include(CheckCXXCompilerFlag)
check_cxx_compiler_flag("-std=c++17" SUPPORT_CXX17) 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") set(dgl_cuda_arch_ptx "70")
if (CUDA_VERSION_MAJOR GREATER_EQUAL "11") if (CUDA_VERSION_MAJOR GREATER_EQUAL "11")
list(APPEND dgl_known_gpu_archs "80") list(APPEND dgl_known_gpu_archs "80" "86")
set(dgl_cuda_arch_ptx "80") set(dgl_cuda_arch_ptx "80" "86")
endif() endif()
# CMake 3.5 doesn't support VERSION_GREATER_EQUAL # CMake 3.5 doesn't support VERSION_GREATER_EQUAL
if (NOT CUDA_VERSION VERSION_LESS "11.8") if (NOT CUDA_VERSION VERSION_LESS "11.8")
......
script/create_dev_conda_env.sh
View file @ 19096c6a
#!/bin/bash #!/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 TORCH_VERSION="1.13.0"
readonly PYTHON_VERSION="3.8" readonly PYTHON_VERSION="3.8"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment