安装对应的库

pip install wandb -i https://pypi.tuna.tsinghua.edu.cn/simple

拓展——wandb是什么模块，有什么作用

wandb是一个用于跟踪、可视化和协作机器学习项目的工具。它提供了许多功能，包括实时的指标跟踪、超参数调整、模型的可视化等，有助于机器学习工程师更好地管理和监控他们的实验，并从中获得更多的见解和知识。

wandb的主体组成主要有以下四大模块：

1. 仪表盘：用于跟踪实验、可视化结果。
2. 报告：保存和分享可复制的成果或结论。
3. Sweeps：通过调节超参数来优化模型。
4. Artifacts：数据集和模型版本化，流水线跟踪。

使用wandb，开发者可以记录模型训练过程中指标的变化情况以及超参的设置，然后对输出的结果进行可视化的对比，帮助更好地分析模型在训练过程中的问题，并快速与同事进行团队协作。

总的来说，wandb是一个强大且实用的工具，可以帮助机器学习开发者更有效地进行项目管理和优化。