DCNv2编译过程报错:error: identifier "THCudaBlas_SgemmBatched" is undefined

解决方法:

将pytorch版本降为1.7.0;

# 先卸载原有的torch和torchvision

pip uninstall torch
pip uninstall torchvision

# 安装pytorh1.7.0

pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 -f https://download.pytorch.org/whl/torch_stable.html

再重新编译。

参考:DCNv2+pytorch1.7及以上版本编译报错解决方法

posted @ 2021-12-30 14:25  盛世芳华  阅读(1484)  评论(0编辑  收藏  举报