Skip to content

GitLab

Unverified Commit b1a532ae authored by J-shang's avatar J-shang Committed by GitHub
Browse files

[Common] change version utils log level (#5023)

parent 84350632
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
nni/common/version.py
View file @ b1a532ae
...@@ -25,21 +25,21 @@ try: ...@@ -25,21 +25,21 @@ try:
import torch import torch
TORCH_VERSION = _minor_version_tuple(torch.__version__) TORCH_VERSION = _minor_version_tuple(torch.__version__)
except ImportError: except ImportError:
logging.getLogger(__name__).info("PyTorch is not installed.") logging.getLogger(__name__).debug("PyTorch is not installed.")
TORCH_VERSION = None TORCH_VERSION = None
try: try:
import pytorch_lightning import pytorch_lightning
PYTORCH_LIGHTNING_VERSION = _minor_version_tuple(pytorch_lightning.__version__) PYTORCH_LIGHTNING_VERSION = _minor_version_tuple(pytorch_lightning.__version__)
except ImportError: except ImportError:
logging.getLogger(__name__).info("PyTorch Lightning is not installed.") logging.getLogger(__name__).debug("PyTorch Lightning is not installed.")
PYTORCH_LIGHTNING_VERSION = None PYTORCH_LIGHTNING_VERSION = None
try: try:
import tensorflow import tensorflow
TENSORFLOW_VERSION = _minor_version_tuple(tensorflow.__version__) TENSORFLOW_VERSION = _minor_version_tuple(tensorflow.__version__)
except ImportError: except ImportError:
logging.getLogger(__name__).info("Tensorflow is not installed.") logging.getLogger(__name__).debug("Tensorflow is not installed.")
TENSORFLOW_VERSION = None TENSORFLOW_VERSION = None
# Serialization version check are needed because they are prone to be inconsistent between versions # Serialization version check are needed because they are prone to be inconsistent between versions
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment