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

Prefer tensorflow_gpu over tensorflow when inspecting installed packages

Browse Source
This commit is contained in:
allegroai 2020-02-04 18:00:39 +02:00
parent 0c71889ca5
commit 8c2b36968b
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
trains/backend_interface/task/repo/scriptinfo.py
View File

@ -46,6 +46,10 @@ class ScriptRequirements(object):
new_mapping.update(mapping)
mapping = new_mapping
# HACK: prefer tensorflow_gpu over tensorflow
if 'tensorflow_gpu' in new_mapping:
new_mapping['tensorflow'] = new_mapping['tensorflow_gpu']
return mapping
def get_requirements(self):