"vscode:/vscode.git/clone" did not exist on "3d38c2a93eeb1e40dd0f7032fdf401966bc3fb96"
Unverified Commit ab28ace1 authored by Matthias Fey's avatar Matthias Fey Committed by GitHub
Browse files

Merge pull request #72 from mpariente/cuda_check

Check if CUDA is available before checking CUDA version
parents ed3de958 000b1ac5
......@@ -12,7 +12,7 @@ for library in [
torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
library, [osp.dirname(__file__)]).origin)
if torch.version.cuda is not None: # pragma: no cover
if torch.version.cuda is not None and torch.cuda.is_available(): # pragma: no cover
cuda_version = torch.ops.torch_sparse.cuda_version()
if cuda_version == -1:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment