如何创建CUDA上下文？

Question

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

Answer 1

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

编辑:请注意,从CUDA 5.0开始,似乎上下文建立的启发式略有不同,并且cudaSetDevice()它本身在设备上建立了上下文.因此,cudaFree(0)不再需要显式调用(尽管它不会伤害任何东西).