We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ba37e4e commit b03e75fCopy full SHA for b03e75f
test/smoke_test/smoke_test.py
@@ -72,7 +72,7 @@ def cuda_runtime_error():
72
def smoke_test_cuda(package: str) -> None:
73
if not torch.cuda.is_available() and is_cuda_system:
74
raise RuntimeError(f"Expected CUDA {gpu_arch_ver}. However CUDA is not loaded.")
75
-
+ if torch.cuda.is_available():
76
if torch.version.cuda != gpu_arch_ver:
77
raise RuntimeError(
78
f"Wrong CUDA version. Loaded: {torch.version.cuda} Expected: {gpu_arch_ver}"
0 commit comments