【问题标题】:TypeError: img should be PIL Image. Got <class 'torch.Tensor'> even though using latest pytorch version类型错误:img 应该是 PIL 图像。得到 <class 'torch.Tensor'> 即使使用最新的 pytorch 版本
【发布时间】:2021-04-30 20:51:23
【问题描述】:

我必须在 Anaconda 中创建一个新的 pytorch 环境,我创建了我的代码,它在我的旧环境中运行良好。然后,我使用相同版本的 pytorch 和 cuda 10.1 创建了新环境(但随后更新到 cuda 11,与旧环境相同。当我尝试运行相同的代码时,我收到错误:

TypeError: img 应该是 PIL Image。得到

当尝试对我的张量应用任何转换时,例如,以下代码给了我一个错误:

def randRoll(batch, deg):
rotator = torchvision.transforms.RandomRotation(deg)
batch = rotator(batch)
return batch

什么都没有改变,我不明白为什么我会得到这个。

有什么建议吗?

【问题讨论】:

    标签: python deep-learning pytorch


    【解决方案1】:

    解决了

    结果以某种方式安装了torchvision 0.2.2,而不是最新的0.9.1(我的其他环境使用的)。

    通过使用卸载 torchvision 解决了这个问题

    conda remove torchvision
    

    然后使用 pip 安装 torchvision(使用 conda install 给了我 0.2.2 版)

    pip install torchvision
    

    我还必须使用 pip 重新安装六个。

    【讨论】:

      猜你喜欢
      • 2021-05-24
      • 1970-01-01
      • 2021-07-16
      • 2021-01-26
      • 1970-01-01
      • 2019-11-06
      • 2018-12-03
      • 2021-06-24
      • 2020-11-05
      相关资源
      最近更新 更多