search

MiniBullLab / easy_test

MIT License
0 stars 2 forks source link

pytorch cpu版本升级 #5

Closed foww-0001 closed 2 years ago

foww-0001 commented 2 years ago

主要修改了torch和torchvision的环境，使用torch==1.6.0和torchvision==0.7.0，测试内容（满足上位机运行）：

foww-0001 commented 2 years ago

运行下面脚步安装cpu版本pytorch：

pip3 install https://download.pytorch.org/whl/cpu/torch-1.6.0%2Bcpu-cp36-cp36m-linux_x86_64.whl
MiniBullLab commented 2 years ago

cpu inference测试

foww-0001 commented 2 years ago

inference已测试完毕，可以试一下

foww-0001 commented 2 years ago

具体cpu运行时间如下： classnet: 30ms detnet: 250ms segnet: 1200ms

kingwangxiang commented 2 years ago

测试已经完成。