Skip to content

Commit 1272272

Browse files
committed
relax versioning
1 parent 981731f commit 1272272

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch_scatter/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@
5050
major, minor = int(str(cuda_version)[0:2]), int(str(cuda_version)[3])
5151
t_major, t_minor = [int(x) for x in torch.version.cuda.split('.')]
5252

53-
if t_major != major or t_minor != minor:
53+
if t_major != major:
5454
raise RuntimeError(
5555
f'Detected that PyTorch and torch_scatter were compiled with '
5656
f'different CUDA versions. PyTorch has CUDA version '

0 commit comments

Comments
 (0)