ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-04-25 00:24:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix PY2.7 support for pytorch

Browse Source
This commit is contained in:
allegroai 2021-10-19 10:47:09 +03:00
parent bed94ee431
commit a890e36a36
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
clearml_agent/helper/package/pytorch.py
View File

@ -263,6 +263,9 @@ class PytorchRequirement(SimpleSubstitution):
continue
if len(parts) < 5 or platform_wheel not in parts[4]:
continue
# yes this is for linux python 2.7 support, this is the only python 2.7 we support...
if py_ver and py_ver[0] == '2' and len(parts) > 3 and not parts[3].endswith('u'):
continue
# update the closest matched version (from above)
if not closest_v:
closest_v = v