没有名为“torch.nn.function”的模块

Question

我有包含以下行的 python 文件：

import argparse
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torchvision import datasets, transforms
from torch.autograd import Variable

它会产生错误：

  File "C:\gdrive\python\a.py", line 5, in <module>
import torch.nn.functional as F
ModuleNotFoundError: No module named 'torch.nn.functional'

如何修复该错误？

我已经使用命令安装了 pytorch：

conda install pytorch-cpu torchvision-cpu -c pytorch

Answer 1

您的 PyTorch 版本似乎已过时。Conda-pytorch-cpu最后一次发布是在一年多前，其最新版本的 PyTorch 是1.1.0，而 PyTorch 目前的版本是1.5.0. 该软件包已被放弃。

您应该按照PyTorch - 本地入门上给出的官方说明安装 PyTorch ，选择您想要的版本。在你的情况下，这将是 Conda 与 CUDA None （以获得仅 CPU 版本）。

结果命令是：

conda install pytorch torchvision cpuonly -c pytorch