mmcv/_ext.cpython-38-x86_64-linux-gnu.so: undefined symbol: _ZN3c104cuda20CUDACachingAllocator9alloc
·

报错mmcv,环境背景CUDA=11.6 torch=1.13.0 mmcv-full=1.7.2,版本没有冲突,解决办法:卸载mmcv-full=1.7.2,并选择本地安装:
步骤一:pip uninstall mmcv-full
步骤二:然后根据你的CUDA和PyTorch版本,我的是 CUDA11.6 和 Torch1.13.0
则从这个网站下:
https://download.openmmlab.com/mmcv/dist/cu116/torch1.13.0/index.html

网址根据你的cuda版本和torch调整,如:你的CUDA10.2 和 Torch1.6.0
则从这个网站下:
https://download.openmmlab.com/mmcv/dist/cu102/torch1.6.0/index.html
下载后,进行手动安装:
pip install mmcv_full-1.7.2-cp38-cp38-manylinux1_x86_64.whl
这次就OK了
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐



所有评论(0)