cuda - 如何创建 CUDA 上下文？

Question

如何创建 CUDA 上下文？ CUDA 的第一次调用很慢，我想在启动内核之前创建上下文。

Answer 1

强制建立运行时 API 上下文的规范方法是调用 cudaFree(0)。如果您有多个设备，请调用 cudaSetDevice() 使用您要在其上建立上下文的设备的 ID，然后调用 cudaFree(0) 来建立上下文。

编辑:请注意，从 CUDA 5.0 开始，上下文建立的启发式方法似乎略有不同，cudaSetDevice() 本身会在调用时在设备上建立上下文。因此不再需要显式的 cudaFree(0) 调用(尽管它不会造成任何伤害)。