Commit eff3bd52 authored by rusty1s's avatar rusty1s
Browse files

update2

parent 31a4031d
......@@ -11,7 +11,7 @@ for library in ['_version', '_basis', '_weighting']:
torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
f'{library}_{suffix}', [osp.dirname(__file__)]).origin)
if torch.version.cuda is not None: # pragma: no cover
if torch.cuda.is_available() and torch.version.cuda is not None:
cuda_version = torch.ops.torch_spline_conv.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