ModuleNotFoundError: No module named ‘torchvision.models.mobilenet‘ 解决方法

一、问题

今天在跑一个 Pytorch 编写的深度学习模型的时候,出现了下面的错误:

二、解决方法

pip install cffi -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install cmake -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install mkl -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install pyyaml -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install numpy -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

pip install torchvision==0.4.1 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

到此,能够成功运行 Pytorch 代码了!!!

叶庭云 CSDN认证博客专家 Python 数据可视化 机器学习
微信搜一搜【修炼Python】分享Python爬虫、数据分析、数据可视化、机器学习有关知识和实例;也分享实用的资料教程、软件工具、学习文档和简历模板。发现求知的乐趣,在不断总结和学习中进步。坚持输出优质文章,期待你的关注,一起交流学习,互相成就,与诸君共勉!
已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 书香水墨 设计师:CSDN官方博客 返回首页