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

Try to make sure tensorboard is available when using torch

Browse Source
This commit is contained in:
allegroai 2020-01-13 11:55:55 +02:00
parent affd6050f6
commit 66b251a62b
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -134,6 +134,17 @@ class ScriptRequirements(object):
modules.add('azure_storage_blob', 'trains.storage', 0)
except Exception:
pass
# if we have torch and it supports tensorboard, we should add that as well
# (because it will not be detected automatically)
if 'torch' in modules and 'tensorboard' not in modules:
# noinspection PyBroadException
try:
# see if this version of torch support tensorboard
import torch.utils.tensorboard
import tensorboard
modules.add('tensorboard', 'torch', 0)
except Exception:
pass
return modules, try_imports, local_mods