pytorch 部署centeros系统报“非法指令(吐核)”错误

最近在部署AI服务的过程中遇到一个非常奇怪的问题。

将AI模型部署到centerOS服务器时,系统报“非法指令(吐核)”错误。模型在windows、ubuntu系统上使用pytorch 1.6版本均能跑通,代码本身是没有问题。没办法,只能逐一排查,最终发现加载torch中的某个方法时直接中断,底层libtorch_cpu.so报错。怀疑是系统和torch版本不兼容。

将torch版本降低至1.5版本,终于解决。。。不过1.5版本中有些api没有,需要自己写,有点坑。看来ai部署的服务器最好还是选用ubuntu系统,比较稳定,避免出现各种奇怪的错误。Mark一下。

猜你喜欢

转载自blog.csdn.net/j_river6666/article/details/125936705