gpt4 book ai didi

cuda - 如何创建 CUDA 上下文?

转载 作者:塔克拉玛干 更新时间:2023-11-03 00:00:11 24 4
gpt4 key购买 nike

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

最佳答案

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

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

关于cuda - 如何创建 CUDA 上下文?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/10415204/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com