msamogh/nonechucks

_get_pytorch_version can't deal with +cuda versions

dreamflasher opened this issue · 0 comments

dreamflasher commented 
  File "/home/marcel/.local/lib/python3.8/site-packages/nonechucks/__init__.py", line 27, in <module>
    MAJOR, MINOR = _get_pytorch_version()
  File "/home/marcel/.local/lib/python3.8/site-packages/nonechucks/__init__.py", line 12, in _get_pytorch_version
    major, minor, patch = [int(x) for x in version.split(".")]
  File "/home/marcel/.local/lib/python3.8/site-packages/nonechucks/__init__.py", line 12, in <listcomp>
    major, minor, patch = [int(x) for x in version.split(".")]
ValueError: invalid literal for int() with base 10: '1+cu110'