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

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 

发布了159 篇原创文章 · 获赞 55 · 访问量 36万+

猜你喜欢

转载自blog.csdn.net/northeastsqure/article/details/103791287