Fix torch._six.PY3 not found in newer pytorch versions

[guigarfr]

Newer python versions do not have variable torch._six.PY3 [Issue] which is also a module not meant to be imported in a project.

PY3 only checks that python version is 3, and we can do that directly using sys.