首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >没有名为'torch.nn.functional‘的模块

没有名为'torch.nn.functional‘的模块
EN

Stack Overflow用户
提问于 2020-05-12 02:42:54
回答 1查看 5.8K关注 0票数 1

我有一个包含行的python文件:

代码语言:javascript
运行
复制
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

它会生成错误:

代码语言:javascript
运行
复制
  File "C:\gdrive\python\a.py", line 5, in <module>
import torch.nn.functional as F
ModuleNotFoundError: No module named 'torch.nn.functional'

如何修复这个错误?

我已经使用以下命令安装了pytorch:

代码语言:javascript
运行
复制
conda install pytorch-cpu torchvision-cpu -c pytorch
EN

回答 1

Stack Overflow用户

发布于 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版本)。

生成的命令是:

代码语言:javascript
运行
复制
conda install pytorch torchvision cpuonly -c pytorch
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/61736959

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档