其他分享
首页 > 其他分享> > DCNv2编译过程报错:error: identifier "THCudaBlas_SgemmBatched" is undefined

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

标签:1.7,undefined,THCudaBlas,torch,编译,报错,pip,torchvision
来源: https://www.cnblogs.com/codeit/p/15748607.html