您的位置: 首页  >  文章  >  PyTorch踩坑记录

PyTorch踩坑记录

分类: 文章 • 2023-04-07 09:44:16

  1. 安装PyTorch
    直接进入官网https://pytorch.org/选择版本后复制执行命令
    PyTorch踩坑记录

  2. BrokenPipeError: [Errno 32] Broken pipe
    将num_workers改为0

  3. IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
    将loss.data[0]改为loss.item()

相关推荐