本文共 599 字，大约阅读时间需要 1 分钟。

PyTorch 从 1.2.0 版本开始正式内置 TensorBoard 支持，这意味着我们可以直接使用内置工具进行可视化分析，无需依赖第三方工具。

安装 TensorBoard

在终端中执行以下命令安装必要工具：

conda activate rootpip install tensorboard -i https://pypi.douban.com/simple/

使用 TensorBoard

在 PyTorch 项目中启用 TensorBoard 可视化：

from torch.utils.tensorboard import SummaryWriterwriter = SummaryWriter('runs', 'pytorch_tutorial')

测试安装

在 Jupyter Notebook 中输入以下代码，确保没有错误即可知安装成功：

from torch.utils.tensorboard import SummaryWriterwriter = SummaryWriter('runs', 'pytorch_tutorial')

注意事项

• 确保 PyTorch 版本为 1.2.0 或更高。
• 测试代码运行时请注意数据加载和模型定义是否正确。
• TensorBoard 的可视化功能需要与 PyTorch 的训练或推理过程同步添加 SummaryWriter 记录信息。

转载地址：http://cgjq.baihongyu.com/