mirror of
https://github.com/deepspeedai/DeepSpeed.git
synced 2025-10-21 08:05:15 +08:00
25 lines
784 B
Python
25 lines
784 B
Python
# Copyright (c) Microsoft Corporation.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
# DeepSpeed Team
|
|
|
|
|
|
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
|