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 1913040 commit 77cfe4bCopy full SHA for 77cfe4b
torch_scatter/__init__.py
@@ -47,7 +47,7 @@
47
torch.ops.torch_scatter.gather_coo = gather_coo_placeholder
48
49
cuda_version = torch.ops.torch_scatter.cuda_version()
50
-if torch.cuda.is_available() and cuda_version != -1: # pragma: no cover
+if torch.version.cuda is not None and cuda_version != -1: # pragma: no cover
51
if cuda_version < 10000:
52
major, minor = int(str(cuda_version)[0]), int(str(cuda_version)[2])
53
else:
0 commit comments