首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >Pytorch已使用Conda安装,但在调用时失败

Pytorch已使用Conda安装,但在调用时失败
EN

Stack Overflow用户
提问于 2021-01-27 22:32:53
回答 1查看 950关注 0票数 0

我正在尝试安装使用BERT的pytorch,但是当按照这里的安装说明操作时:https://pytorch.org/get-started/locally/ I遇到一个错误。

当我尝试初始化BERT模型时,我得到了以下错误:

代码语言:javascript
运行
复制
ImportError: 
    BertForSequenceClassification requires the PyTorch library but it was not found in your environment. 
    Checkout the instructions on theinstallation page: https://pytorch.org/get-started/locally/
    and follow the ones that match your environment.

我已经按照说明在Conda提示终端和当前工作目录中运行了以下命令行:

代码语言:javascript
运行
复制
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

它看起来像是完成了,但当我尝试调用以下代码行时,我得到了与start相同的错误,就好像它根本没有安装一样。

有谁能帮帮我吗?

编辑:

我用于执行bert的代码是:

代码语言:javascript
运行
复制
model = BertForSequenceClassification.from_pretrained(r'C:\Users\441\bert\pytorch_model.bin', config = r'C:\Users\441\bert\config.json')
EN

回答 1

Stack Overflow用户

发布于 2021-06-14 22:57:48

我也有同样的问题(同样的错误消息),在使用conda list | grep torch之后,我也发现它就在那里。对我来说起作用的是,我重新启动了jupyter笔记本内核,错误消失了。

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/65921244

复制
相关文章

相似问题

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