fix ValueError: invalid literal for int() with base 10

[gabrielraya]

I got this error

Apparently the fix is done by changing :

• torch_version = [int(s) for s in torch.version.split(".")]

by

• torch_version = [int(s) for s in torch.version.replace("+",".").split('.') if s.isdigit()]

[sdatkinson]

How is torch being installed?

Also, it seems that the real underlying issue is that we're using more information than we need to in order to figure out what potri is being called by PyTorch. If all we grabbed was the major and minor version eg

_torch_majorminor = [int(s) for s in torch.__version__.split(".")[:2]]
_potri = torch.cholesky_inverse if _torch_majorminor >= [1, 1] else torch.potri

then we wouldn't need to hack the CUDA bit on the end.

[nanxu2]

Hi Steven,
I've got the same issue.
My python version is 3.8 with torch 1.10.0.

[sdatkinson]

Should be fixed in main now. @nanxu2 @gabrielraya go ahead and try it out and let me know if it fixes the issue (I can only guess the root cause is you're installing w/ e.g. pip install torch=1.10.0+cpu instead of the plain pip install torch).

I'll put this to PyPI in a new patch version if it fixes the issue.

[nanxu2]

@sdatkinson Hi Steven, It fixed the issue. The code can properly work on my desktop now. Thank you so much.

[sdatkinson]

Glad to hear it :)