Spaces:
Runtime error
Runtime error
diff --git a/setup.py b/setup.py | |
index 5f68ecf..b4d44a8 100644 | |
--- a/setup.py | |
+++ b/setup.py | |
print("\nCompiling cuda extensions with") | |
print(raw_output + "from " + cuda_dir + "/bin\n") | |
- if (bare_metal_major != torch_binary_major) or (bare_metal_minor != torch_binary_minor): | |
- raise RuntimeError( | |
- "Cuda extensions are being compiled with a version of Cuda that does " | |
- "not match the version used to compile Pytorch binaries. " | |
- "Pytorch binaries were compiled with Cuda {}.\n".format(torch.version.cuda) | |
- + "In some cases, a minor-version mismatch will not cause later errors: " | |
- "https://github.com/NVIDIA/apex/pull/323#discussion_r287021798. " | |
- "You can try commenting out this check (at your own risk)." | |
- ) | |
+# if (bare_metal_major != torch_binary_major) or (bare_metal_minor != torch_binary_minor): | |
+# raise RuntimeError( | |
+# "Cuda extensions are being compiled with a version of Cuda that does " | |
+# "not match the version used to compile Pytorch binaries. " | |
+# "Pytorch binaries were compiled with Cuda {}.\n".format(torch.version.cuda) | |
+# + "In some cases, a minor-version mismatch will not cause later errors: " | |
+# "https://github.com/NVIDIA/apex/pull/323#discussion_r287021798. " | |
+# "You can try commenting out this check (at your own risk)." | |
+# ) | |
def raise_if_cuda_home_none(global_option: str) -> None: | |