【发布时间】:2021-01-26 23:29:21
【问题描述】:
我正在尝试让 pytorch 工作,但我不断收到此错误。
ModuleNotFoundError:没有名为“torch.nn”的模块; “火炬”不是一个包
我使用的是 Macbook,我已尝试查看此处的其他答案,但没有任何效果。
import torch
import torchvision
import torchvision.transforms as transforms
transform = transforms.Compose(
[transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
download=True, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
shuffle=True, num_workers=2)
testset = torchvision.datasets.CIFAR10(root='./data', train=False,
download=True, transform=transform)
testloader = torch.utils.data.DataLoader(testset, batch_size=4,
shuffle=False, num_workers=2)
classes = ('plane', 'car', 'bird', 'cat',
'deer', 'dog', 'frog', 'horse', 'ship', 'truck')
【问题讨论】:
标签: python pytorch torchvision