Skip to content

GitLab

Unverified Commit e8473822 authored by Masaki Kozuki's avatar Masaki Kozuki Committed by GitHub
Browse files

version guard (#1253)

parent e0f5ea8c
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
setup.py
View file @ e8473822
...@@ -52,7 +52,10 @@ def raise_if_cuda_home_none(global_option: str) -> None: ...@@ -52,7 +52,10 @@ def raise_if_cuda_home_none(global_option: str) -> None:
def append_nvcc_threads(nvcc_extra_args): def append_nvcc_threads(nvcc_extra_args):
return nvcc_extra_args + ["--threads", "4"] _, bare_metal_major, bare_metal_minor = get_cuda_bare_metal_version(CUDA_HOME)
if int(bare_metal_major) >= 11 and int(bare_metal_minor) >= 2:
return nvcc_extra_args + ["--threads", "4"]
return nvcc_extra_args
if not torch.cuda.is_available(): if not torch.cuda.is_available():
......
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