diff --git a/README.md b/README.md index c49f6e7f7..c96692c60 100644 --- a/README.md +++ b/README.md @@ -54,6 +54,8 @@ To install the remaining basic dependencies, run: ```bash pip install -r requirements/requirements.txt +pip install -r requirements/requirements-wandb.txt +pip install -r requirements/requirements-tensorboard.txt python ./megatron/fused_kernels/setup.py install # optional if not using fused kernels ``` diff --git a/requirements/requirements-tensorboard.txt b/requirements/requirements-tensorboard.txt index fef3fbbc7..6d5967364 100644 --- a/requirements/requirements-tensorboard.txt +++ b/requirements/requirements-tensorboard.txt @@ -1 +1 @@ -tensorboard==2.5.0 +tensorboard==2.13.0