We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 981731f commit 1272272Copy full SHA for 1272272
torch_scatter/__init__.py
@@ -50,7 +50,7 @@
50
major, minor = int(str(cuda_version)[0:2]), int(str(cuda_version)[3])
51
t_major, t_minor = [int(x) for x in torch.version.cuda.split('.')]
52
53
- if t_major != major or t_minor != minor:
+ if t_major != major:
54
raise RuntimeError(
55
f'Detected that PyTorch and torch_scatter were compiled with '
56
f'different CUDA versions. PyTorch has CUDA version '
0 commit comments