Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "ed6983ea843590c05593fcd4c8de40f5c7bb0970"
Commit 124bc097 authored by rusty1s's avatar rusty1s
Browse files

fix cuda context

parent 2da6195d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torch_sparse/__init__.py
View file @ 124bc097
...@@ -22,7 +22,7 @@ for library in [ ...@@ -22,7 +22,7 @@ for library in [
f"{osp.dirname(__file__)}") f"{osp.dirname(__file__)}")
cuda_version = torch.ops.torch_sparse.cuda_version() cuda_version = torch.ops.torch_sparse.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: if cuda_version < 10000:
major, minor = int(str(cuda_version)[0]), int(str(cuda_version)[2]) major, minor = int(str(cuda_version)[0]), int(str(cuda_version)[2])
else: else:
......
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