AI老铁 · 2020年12月14日

双目测距系列(五)Win10下面安装pytorch

转载自:双目测距系列(五)Win10下面安装pytorch
作者:ltshan139

前言

在window上安装pytorch虽然比caffe容易得多,但是也遇到了各种各样得坑,这里我把自己遇到的坑记录分享一下。

DLL加载不成功的坑

起初,我的python版本是3.5,因为网络的原因,直接 pip install torch torchvision很难成功。 所以下载torch-1.0.0-cp35-cp35m-win_amd64.whl,手动安装如下:

pip install torch-1.0.0-cp35-cp35m-win_amd64.whl
可是安装成功后 去import torch时遇到了:

from torch._C import *
ImportError: DLL load failed: 找不到指定的模块
网上有很多种解决方法,比如说 pip install intel-openmp或者添加python安装目录下torch的lib路径到系统path,但我都试过了没有用。

解决办法

因为我之前的python版本是3.5.1,所以我怀疑是因为python版本低引起,所以又装了python3.6,并且将其安装路径添加到系统路径,确保在cmd下输入pytho命令,其对应版本为3.6。

然后在pytorch官网选取稳定版本:

image.png

将上面的command粘贴到cmd窗口:
image.png

结论

当遇到torch C dll无法加载的错误时,可以考虑将python版本升级来解决。



推荐阅读



更多海思AI芯片方案学习笔记欢迎关注海思AI芯片方案学习

推荐阅读
关注数
18835
内容数
1369
嵌入式端AI,包括AI算法在推理框架Tengine,MNN,NCNN,PaddlePaddle及相关芯片上的实现。欢迎加入微信交流群,微信号:aijishu20(备注:嵌入式)
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息