Skip to content

GitLab

Unverified Commit 2b7d3aca authored by Cody Yu's avatar Cody Yu Committed by GitHub
Browse files

Update setup.py (#282) 


Co-authored-by: default avatarneubig <neubig@gmail.com>
parent 4026a049
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
setup.py
View file @ 2b7d3aca
......@@ -20,10 +20,9 @@ ABI = 1 if torch._C._GLIBCXX_USE_CXX11_ABI else 0
CXX_FLAGS += [f"-D_GLIBCXX_USE_CXX11_ABI={ABI}"]
NVCC_FLAGS += [f"-D_GLIBCXX_USE_CXX11_ABI={ABI}"]
if not torch.cuda.is_available():
if CUDA_HOME is None:
raise RuntimeError(
f"Cannot find CUDA at CUDA_HOME: {CUDA_HOME}. "
"CUDA must be available in order to build the package.")
f"Cannot find CUDA_HOME. CUDA must be available in order to build the package.")
def get_nvcc_cuda_version(cuda_dir: str) -> Version:
......
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