雨滴科技技术论坛

 找回密码
 立即注册
查看: 534|回复: 2

torchvision安装有一个坑

[复制链接]

513

主题

1974

帖子

8158

积分

论坛元老

Rank: 8Rank: 8

积分
8158
扫一扫,手机访问本帖
发表于 2021-3-5 21:07:09 | 显示全部楼层 |阅读模式
在x86的主机上面,一切交给anaconda来解决。一句命令就可以安装。

在arm主机上面,需要aarch64架构的安装包。用pip3找不到对应的wheel

pytorch的可以下载得到1.7.0的wheel
pip3 install  wheel的方式可以安装成功

torchvision需要从github下载代码来安装。

选择了版本0.8.1之后
sudo python3 setup.py install
安装也成功了

但是在依赖检查时候一直报错说是没有正确的版本。

弄了很久,最后找到原因,打开setup.py
看到里面的版本是0.8.0a0这是代码的坑,导致版本一直对不上。

手工修改版本号0.8.1再来一次,成功解决依赖。
回复

使用道具 举报

513

主题

1974

帖子

8158

积分

论坛元老

Rank: 8Rank: 8

积分
8158
 楼主| 发表于 2021-3-5 21:11:35 | 显示全部楼层
yolov5的small模型, 在没有使用tensorrt加速情况下, 512xx640分辨率, 大概0.18s一帧, 也就是才5帧多的速度。
回复 支持 反对

使用道具 举报

513

主题

1974

帖子

8158

积分

论坛元老

Rank: 8Rank: 8

积分
8158
 楼主| 发表于 2021-3-5 21:15:44 | 显示全部楼层
1280x720分辨率。
ssd + mobilenet的用tensorrt加速,达到22fps
回复 支持 反对

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

在线客服
在线咨询
咨询热线
0755-26787502-8006/8016
扫一扫二维码
直接访问本站

QQ|Archiver|手机版|小黑屋|雨滴科技  

GMT+8, 2021-10-22 04:54 , Processed in 0.052988 second(s), 11 queries , Gzip On, Memcache On.

Powered by Discuz! X3.2

© 2001-2013 Comsenz Inc.

快速回复 返回顶部 返回列表