diff --git a/mmcv/runner/hooks/logger/tensorboard.py b/mmcv/runner/hooks/logger/tensorboard.py index ad13395f7..63058e8dd 100644 --- a/mmcv/runner/hooks/logger/tensorboard.py +++ b/mmcv/runner/hooks/logger/tensorboard.py @@ -22,7 +22,7 @@ class TensorboardLoggerHook(LoggerHook): @master_only def before_run(self, runner): - if torch.__version__ >= '1.1': + if torch.__version__ >= '1.1' and '.' in torch.__version__: try: from torch.utils.tensorboard import SummaryWriter except ImportError: