【问题标题】:ModuleNotFoundError: No module named 'torch.nn'; 'torch' is not a package on Mac OSModuleNotFoundError:没有名为“torch.nn”的模块; “火炬”不是 Mac OS 上的软件包
【发布时间】: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


    【解决方案1】:

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

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

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

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 2021-06-08
      • 2021-09-06
      • 2022-10-24
      • 2021-08-26
      • 2018-03-28
      • 2022-06-17
      • 1970-01-01
      相关资源
      最近更新 更多