我有一个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
2条答案
按热度按时间w1e3prcc1#
看起来你有一个过时的PyTorch版本。Conda -
pytorch-cpu最后一次发布是在一年多前，其最新版本的PyTorch是1.1.0，而PyTorch目前的版本是1.5.0。包裹已经被丢弃了。你应该按照PyTorch - Get Started locally上的官方说明安装PyTorch，选择你想要的版本。在您的情况下，这将是Conda与CUDA无（以获得CPU唯一的版本）。
生成的命令为：
hujrc8aj2#
试试这个
from torch.nn import functional as F