Unverified Commit ff58fa7e authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[build] build against installed cuda-11.1 while torch built w/ cuda-11.0 (#570)

parent be33bea4
...@@ -44,6 +44,9 @@ def assert_no_cuda_mismatch(): ...@@ -44,6 +44,9 @@ def assert_no_cuda_mismatch():
torch_cuda_version = ".".join(torch.version.cuda.split('.')[:2]) torch_cuda_version = ".".join(torch.version.cuda.split('.')[:2])
# This is a show-stopping error, should probably not proceed past this # This is a show-stopping error, should probably not proceed past this
if sys_cuda_version != torch_cuda_version: if sys_cuda_version != torch_cuda_version:
if sys_cuda_version == "11.1" and torch_cuda_version == "11.0":
# it works to build against installed cuda-11.1 while torch was built with cuda-11.0
return
raise Exception( raise Exception(
f"Installed CUDA version {sys_cuda_version} does not match the " f"Installed CUDA version {sys_cuda_version} does not match the "
f"version torch was compiled with {torch.version.cuda}, unable to compile " f"version torch was compiled with {torch.version.cuda}, unable to compile "
......
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