Pytorch
yy迷小迭
心向阳光,明媚而生
展开
-
TX2配置pytorch环境
TX2系统 Jetpack4.6版本 1、nvidia官方下载pytorch1.6版本轮子 2、安装pytorch pip3 install Cython pip3 install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.whl 3、安装相关依赖 sudo apt-get install libjpeg-dev zlib1g-dev 4、安装torchvision 版本匹配 git clone --branch v0.7.0-rc2 https://git原创 2021-09-24 11:08:10 · 397 阅读 · 0 评论 -
Pytorch模型部署开发板所遇到的坑.....
工作原因,需要将pytorch模型部署到rk3399。 现一步步记录下这些bug哈(晕)----希望最后能把这项目完成吧 pytorch ----> onnx Bug 1: RuntimeError: Failed to export an ONNX attribute, since it’s not constant, please try to make things (e.g., kernel size) static if possible 解决方案: 一开始看blog说是pytorc原创 2020-08-19 11:08:41 · 882 阅读 · 5 评论