ModuleNotFoundError: 没有名为“torch.nn”的模块； 'torch' 不是 Mac OS 上的软件包

Question

我正在尝试让 pytorch 工作，但我不断收到此错误。

ModuleNotFoundError: 没有名为“torch.nn”的模块； 'torch' 不是一个包

我使用的是 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')

Answer 1

也许您可以检查 conda list 以查看是否安装了 PyTorch。如果您安装了 PyTorch，您应该可以运行 Torch。

下载链接：https://pytorch.org/get-started/locally/

如果您想使用 GPU 而不是 CPU，请记住另外安装 CUDA。