【发布时间】: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