记一个pytorch和mxne共同载入，程序退出时候，unload报错

最新推荐文章于 2023-05-28 08:06:45 发布

northeastsqure

最新推荐文章于 2023-05-28 08:06:45 发布

阅读量279

点赞数

文章标签： mxnet pytorch

本文链接：https://blog.csdn.net/northeastsqure/article/details/103791287

版权

Segmentation fault: 11 

Stack trace: 

  [bt] (0) /home/silva/anaconda3/envs/py372/lib/python3.7/site-packages/mxnet/libmxnet.so(+0x2e6b140) [0x7f0e63059140] 

  [bt] (1) /lib/x86_64-linux-gnu/libc.so.6(+0x354b0) [0x7f0f5be634b0] 

  [bt] (2) /home/silva/anaconda3/envs/py372/lib/python3.7/site-packages/torchvision/../../../libcudart.so.9.2(+0x17dc4) [0x7f0efa636dc4] 

  [bt] (3) /home/silva/anaconda3/envs/py372/lib/python3.7/site-packages/torchvision/../../../libcudart.so.9.2(+0x1268d) [0x7f0efa63168d] 

  [bt] (4) /home/silva/anaconda3/envs/py372/lib/python3.7/site-packages/torchvision/../../../libcudart.so.9.2(cudaSetDevice+0x5e) [0x7f0efa661e0e] 

  [bt] (5) /home/silva/anaconda3/envs/py372/lib/python3.7/site-packages/mxnet/libmxnet.so(+0x25bf40a) [0x7f0e627ad40a] 

  [bt] (6) /home/silva/anaconda3/envs/py372/lib/python3.7/site-packages/mxnet/libmxnet.so(+0x25cc21e) [0x7f0e627ba21e] 

  [bt] (7) /home/silva/anaconda3/envs/py372/lib/python3.7/site-packages/mxnet/libmxnet.so(+0x25b4d8e) [0x7f0e627a2d8e] 

  [bt] (8) /home/silva/anaconda3/envs/py372/lib/python3.7/site-packages/mxnet/libmxnet.so(+0x25b5cd4) [0x7f0e627a3cd4]

解决办法是：

原因是torch内部带了

python3.7/site-packages/torch/lib/libcudart-1b201d85.so.10.1，所以torch和mxnet 不同版本,一起载入后，在两者unload的时候，mxnet报错了，解决办法是，mxnet也安装mxnet-101