当前位置:   article > 正文

ImportError: /usr/local/anaconda3/envs/py38/lib/python3.8/site-packages/mmcv/_ext.cpython-38-x86_64-_/home/gao2/anaconda3/envs/xiaojie_py38/lib/python3

/home/gao2/anaconda3/envs/xiaojie_py38/lib/python3.8/site-packages/mmcv/imag

在安装mmcv-full后,进行单机多卡分布式训练时遇到的问题,主要有两个:

在这里插入图片描述

ImportError: /usr/local/anaconda3/envs/py38/lib/python3.8/site-packages/mmcv/_ext.cpython-38-x86_64-linux-gnu.so: undefined symbol: Z27points_in_boxes_cpu_forwardN2at6TensorES0_S0

这个是错误之一,有时可能还是:
ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory

mmcv官方:https://mmcv.readthedocs.io/en/latest/get_started/installation.html

这些问题归根到底还是mmcv-full和自己安装的pytorch版本不匹配的原因导致的,所以在安装mmcv-full时,尽量不要偷懒,不要直接pip install mmcv-full,先检查好自己的nvidia驱动以及CUDA版本,然后去mmcv的github或者官网去寻找指定的命令去下载,找到匹配的版本下载安装后,就可以正常运行了。

这是通过mmediting调用mmcv正常运行的截图:
在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/羊村懒王/article/detail/313022
推荐阅读
相关标签
  

闽ICP备14008679号