From 1df223e02bf0f8a7a49ebc8db4226f7cb811a67d Mon Sep 17 00:00:00 2001 From: rusty1s Date: Fri, 25 Mar 2022 09:20:21 +0000 Subject: [PATCH] prevent creating cuda context on import --- torch_scatter/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_scatter/__init__.py b/torch_scatter/__init__.py index 8fff3818..ef5a4271 100644 --- a/torch_scatter/__init__.py +++ b/torch_scatter/__init__.py @@ -47,7 +47,7 @@ torch.ops.torch_scatter.gather_coo = gather_coo_placeholder cuda_version = torch.ops.torch_scatter.cuda_version() -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 if cuda_version < 10000: major, minor = int(str(cuda_version)[0]), int(str(cuda_version)[2]) else: