grid_sample_3d_cuda 中的错误：没有可在设备上执行的内核映像

Question

当我运行tensorrt(8.6.1)来加速SadTalker时

grid_sample_3d_cuda 中的错误：没有可在设备上执行的内核映像 grid_sample_3d_cuda 中出现错误：没有可在设备上执行的内核映像

Answer 1

我刚刚遇到了同样的问题。查了资料发现是编译grid_sample_3d插件时CUDA架构设置不正确导致的。

将编译好的插件的CMakeLists.txt中的

set_target_properties(${PROJECT_NAME} PROPERTIES CUDA_ARCHITECTURES “89”)

设置为适合自己GPU的算力。计算能力可以在 URL https://developer.nvidia.com/cuda-gpus 找到。例如，如果我使用的是 3090，对应的计算能力为 8.6，我应该使用

set_target_properties(${PROJECT_NAME} PROPERTIES CUDA_ARCHITECTURES “86”)