我有一个包含行的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
发布于 2020-05-12 09:19:00
看起来你有一个过时的PyTorch版本。Conda - pytorch-cpu
上一次发布是在一年前,其最新的PyTorch版本是1.1.0
,而PyTorch目前的版本是1.5.0
。那个包裹已经被丢弃了。
您应该按照PyTorch - Get Started locally上给出的官方说明安装PyTorch,方法是选择所需的版本。在您的例子中,应该是Conda with CUDA None (以获得仅CPU版本)。
生成的命令是:
conda install pytorch torchvision cpuonly -c pytorch
https://stackoverflow.com/questions/61736959
复制相似问题