pytorch网络模型运行不了

[复制链接] · 发表于 2021-6-18 15:13:19

本帖最后由 chenshq168 于 2021-6-18 17:03 编辑

一个奇怪的问题运行toolkit里面的示例程序时，提示以下错误：
examples/pytorch/resnet18$ python3 test.py
/home/toybrick/.local/lib/python3.7/site-packages/torch/package/_mock_zipreader.py:17: UserWarning: Failed to initialize NumPy: module compiled against API version 0xe but this version of numpy is 0xd (Triggered internally at ../torch/csrc/utils/tensor_numpy.cpp:67.)
_dtype_to_storage = {data_type(0).dtype: data_type for data_type in _storages}
Illegal instruction

如果把numpy升级到1.20.3版本时，又出现这样的错误：

examples/pytorch/resnet18$ python3 test.py
Illegal instruction

最郁闷的是根本上不知道是什么原因造成的。原厂能不能多开放一些东东让我们看看呢？

只看该作者 · 发表于 2021-6-18 18:51:41

网络模型跟哪个版本相关，至少也出个readme嘛，搞得一个版本一版本的去试，让人崩溃。

只看该作者 · 发表于 2021-7-3 10:41:34

升级Torch版本，1.5.1或者1.6.0，应该能解决