问我如何告诉PyCUDA使用哪个GPU？

EN

nvcc不会根据您安装的特定GPU来抱怨。它将编译任何您告诉它要编译的GPU类型。问题是，当还指定了一个sm_30选项时，您指定的是--gpu-architecture的非有效选项--gpu-code。

您应该将compute_30传递给--gpu-architecture，sm_30传递给--gpu-code

另外，请确保您使用的是正确的nvcc，并且不会无意中使用一些旧版本的CUDA工具包。

一旦解决了编译问题，就会有一个环境变量CUDA_DEVICE，pycuda将观察到这个变量来选择特定安装的GPU。

来自这里

CUDA_DEVICE=2 python my-script.py

顺便说一句，其他人有你的问题。你确定你没有一个旧版本的PyCUDA正在使用的CUDA工具包吗？

我不知道Python包装器(或者一般的Python )，但是在C++中有亲和力 NVidia扩展，它允许您针对特定的GPU。也许您可以用Python为它编写一个包装器。

编辑：

现在我看到您实际上正在运行Linux，解决方案是更简单的(C++).You只需要在上下文init之前枚举XDisplay。

因此，默认的GPU通常以显示字符串"0.0“为目标。

要用第二个GPU打开显示器，您可以这样做：

    const char* gpuNum = "0:1";

    if (!(_display = XOpenDisplay(gpuNum ))) {

        printf("error: %s\n", "failed to open display");

    } else {
        printf("message: %s\n", "display created");

    }

       ////here comes the rest of context setup....

至少目前看来，我们可以说

import pycuda.driver as drv
drv.Device(6).make_context()

还有这个将设备6设置为当前上下文。