- 博客(1)
- 收藏
- 关注
原创 Deformable DETR 运行make.sh编译CUDA算子报错
1. 在conda环境中下载和系统cuda版本(nvcc -V查看版本,本人的是11.5)差不多版本的cudatoolkit。(nvidia-smi查看的是12.1版本)print(torch.version.cuda)"查看conda环境下调用的cuda版本是否正确。注:修改CUDA_HOME、LD_LIBRARY_PATH等没有用,如果指定为conda环境中的路径,编译时会报错 nvcc not found。2. 降低gcc版本,本人原版本是11,将为9。
2025-03-18 16:14:20
165
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人