File size: 1,226 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
# Copyright (c) Microsoft Corporation.
# SPDX-License-Identifier: Apache-2.0
# DeepSpeed Team
from packaging import version as pkg_version
def check_tb_availability():
try:
# torch.utils.tensorboard will fail if `tensorboard` is not available,
# see their docs for more details: https://pytorch.org/docs/1.8.0/tensorboard.html
import tensorboard # noqa: F401 # type: ignore
except ImportError:
print('If you want to use tensorboard logging, please `pip install tensorboard`')
raise
def check_wandb_availability():
try:
import wandb # noqa: F401 # type: ignore
except ImportError:
print(
'If you want to use wandb logging, please `pip install wandb` and follow the instructions at https://docs.wandb.ai/quickstart'
)
raise
def check_comet_availability():
try:
import comet_ml
comet_version = pkg_version.parse(comet_ml.__version__)
if comet_version < pkg_version.Version("3.41.0"):
raise ImportError("`comet_ml` must have at least version 3.41.0")
except ImportError:
print('If you want to use comet logging, please `pip install "comet_ml>=3.41.0"`')
raise
|