- 博客(3)
- 收藏
- 关注
原创 MultiScaleDeformableAttention编译问题:no kernel image is available for execution on the device
因为我的GPU是 NVIDIA RTX A6000,它的算力是8.6(Compute Capability 8.6,对应 sm_86),所以最低需要CUDA 11.0及以上版本的支持,建议安装CUDA 11.3或11.4、11.7等更高版本。
2025-10-28 10:48:22
300
原创 编译MultiScaleDeformableAttention时,出现RuntimeError: Error compiling objects for extension问题
由于我的 cusolverDn.h 实际在 /usr/local/cuda-11.1/targets/x86_64-linux/include/ 不是标准的 /usr/local/cuda/include,则还需要在编译时候加上 include 路径。一般建议使用官方 CUDA Toolkit,而不是只依赖 conda 虚拟环境的 cudatoolkit,因为 conda 里的 cudatoolkit 不含开发头文件,仅供 PyTorch 运行时调用,不支持 C++/CUDA 的源码编译扩展!
2025-10-28 10:33:01
773
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅