ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-01-31 09:06:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Change message when using local torch

Browse Source
This commit is contained in:
allegroai 2020-03-20 10:37:42 +02:00
parent 4158d08f6f
commit 94bb11a81a
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
trains_agent/helper/package/pytorch.py
View File

@ -254,9 +254,10 @@ class PytorchRequirement(SimpleSubstitution):
if self.config.get("agent.package_manager.system_site_packages"): if self.config.get("agent.package_manager.system_site_packages"):
from pip._internal.commands.show import search_packages_info from pip._internal.commands.show import search_packages_info
installed_torch = list(search_packages_info([req.name])) installed_torch = list(search_packages_info([req.name]))
op, version = req.specs[0] if req.specs else (None, None)
# notice the comparision order, the first part will make sure we have a valid installed package # notice the comparision order, the first part will make sure we have a valid installed package
if installed_torch[0]['version'] and (installed_torch[0]['version'] == version or not version): if installed_torch[0]['version'] and req.compare_version(installed_torch[0]['version']):
print('PyTorch: requested "{}" version {}, using pre-installed version {}'.format(
req.name, req.specs[0] if req.specs else 'unspecified', installed_torch[0]['version']))
# package already installed, do nothing # package already installed, do nothing
return str(req), True return str(req), True
except: except: