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及以上版本编译报错解决方法

上一篇:pytorch 学习


下一篇:torch 单机多卡运行