fix cuda context

Author: rusty1s

torch_cluster/__init__.py

@@ -21,7 +21,7 @@
                           f"{osp.dirname(__file__)}")
 
 cuda_version = torch.ops.torch_cluster.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: