
树莓派
全宇宙最最帅气的哆啦A梦小怪兽
前端学习中,成长型博主,努力学习总结中……
展开
-
使用torch.load()加载模型参数时,提示“xxx.pt is a zip archive(did you mean to use torch.jit.load()?)“
原因xxx.pth来自pytorch1.6或更高的版本。1.6之后pytorch默认使用zip文件格式来保存权重文件,导致这些权重文件无法直接被1.5及以下的pytorch加载。解决方案在含有高版本的torch的设备上运行如下代码:import torchstate_dict = torch.load("E:/FinalDesign/训练结果/shuffleNet/fruitssm622 shufflenet epoch=40/ShuffleNetV2.pth",map_location='cpu原创 2022-05-08 22:08:48 · 533 阅读 · 0 评论 -
树莓派4B安装pytorch
树莓派4B4G版python3.7下载安装包torch链接:https://pan.baidu.com/s/1i0ZCwOd7YCTtimPbFz_WsQ 提取码:dc4ztorchvision链接:https://pan.baidu.com/s/1d_cxeosu1DkYH5kS8r35aA 提取码:8zgy安装Pytorch安装必要的依赖包sudo apt-get install libopenblas-dev安装numpy pip3 install numpy pyyaml(没有nump原创 2022-03-09 17:47:57 · 1098 阅读 · 0 评论