当“cuMemGetAddressRange”存在时Cuda编译错误[重复]

Question

示例代码如下：

#include <cuda.h>
#include <cuda_runtime.h>
#include <stdio.h>

int main() {
    unsigned char* cu_test;
    cudaMalloc((void**)&cu_test, 3200);

    CUdeviceptr pbase;
    size_t psize;
    CUresult res = cuMemGetAddressRange(&pbase, &psize, (CUdeviceptr)cu_test);
    printf("cu_img_yuv size: %ld", psize);

    return 0;
}

虽然在编译任何 cuda 版本时都会引发错误（从 11.3 到 11.5 测试）：

$ nvcc main.cu -o main
/tmp/tmpxft_0000e288_00000000-11_main.o: In function `main':
tmpxft_0000e288_00000000-6_main.cudafe1.cpp:(.text+0x54): undefined reference to `cuMemGetAddressRange_v2'
collect2: error: ld returned 1 exit status

有人可以帮忙指出问题所在吗？

Answer 1

消息：

error: ld returned 1 exit status

表示这是一个链接错误（参见ld）。

要查看是这种情况，您可以运行

nvcc -c main.cu -o main.o

你不会得到任何错误。这是源代码编译步骤！

解决方案：

您需要明确指定与 CUDA 驱动存根库的链接：

nvcc main.cu -o main -lcuda

这是因为cuMemGetAddressRange() 是 CUDA 驱动程序 API 的一部分，而不是 CUDA 运行时 API。

注意：您没有cudaFree() 分配的内存，您可能需要修复此问题！

编辑：（归功于@talonmies 评论）您不必显式链接到 CUDA 运行时库 (-lcudart)，因为 nvcc 会自动链接到它。