欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 健康 > 美食 > 【已解决】: fatal error: cuda_runtime_api.h: No such file or directory

【已解决】: fatal error: cuda_runtime_api.h: No such file or directory

2024/10/26 8:24:45 来源:https://blog.csdn.net/qq_51219814/article/details/140184192  浏览:    关键词:【已解决】: fatal error: cuda_runtime_api.h: No such file or directory

既然他找不到,我们就把路径给他写清楚!

  1. 检查自己是不是有这个文件:
  • 去路径/usr/local下,使用命令查询是否拥有该文件:
find . -name cuda_runtime_api.h

结果:
在这里插入图片描述
因为我要使用的是cuda-11.3,因此复制该路径

  1. 将该路径粘贴加入setup.py文件里:
include_dirs = [os.path.realpath('../include', '/usr/local/cuda-11.3/targets/x86_64-linux/include'])CUDAExtension('roi_align.crop_and_resize_gpu',['roi_align/src/crop_and_resize_gpu.cpp','roi_align/src/cuda/crop_and_resize_kernel.cu'],include_dirs=include_dirs,#添加这一行!!!!extra_compile_args={'cxx': ['-g', '-fopenmp'],'nvcc': ['-O2']})

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com