Commit 8591256d authored by rusty1s's avatar rusty1s
Browse files

relaxed CUDA version req

parent 7fc01df2
...@@ -23,7 +23,7 @@ if torch.version.cuda is not None: # pragma: no cover ...@@ -23,7 +23,7 @@ if torch.version.cuda is not None: # pragma: no cover
major, minor = int(str(cuda_version)[0:2]), int(str(cuda_version)[3]) major, minor = int(str(cuda_version)[0:2]), int(str(cuda_version)[3])
t_major, t_minor = [int(x) for x in torch.version.cuda.split('.')] t_major, t_minor = [int(x) for x in torch.version.cuda.split('.')]
if t_major != major or t_minor != minor: if t_major != major:
raise RuntimeError( raise RuntimeError(
f'Detected that PyTorch and torch_cluster were compiled with ' f'Detected that PyTorch and torch_cluster were compiled with '
f'different CUDA versions. PyTorch has CUDA version ' f'different CUDA versions. PyTorch has CUDA version '
......
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