在编译deformable注意力的时候
这是pytorch版本和torchvision不匹配
官方编译cuda是torch=1.5.1，torchvision=0.6.1
应该进入torch官网选择
conda install pytorch1.7.1 torchvision0.8.2 torchaudio==0.7.2 cudatoolkit=9.2 -c pytorch
就可以编译cuda了
问题：_ZN6caffe28TypeMeta21_typeMetaDataInstanceIN3c107complexINS2_4HalfEEEEEPKNS_
最新推荐文章于 2023-08-31 20:05:22 发布
在尝试编译deformable注意力模块时遇到了PyTorch与TorchVision版本不一致的问题。官方建议使用torch=1.5.1和torchvision=0.6.1来配合CUDA。为解决此问题，可以按照官方提示，通过conda安装指定版本：`conda install pytorch=1.7.1 torchvision=0.8.2 torchaudio==0.7.2 cudatoolkit=9.2 -c pytorch`，这将确保与CUDA的兼容性。
扫一扫
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
