当前位置: 首页> 房产> 政策 > 【教程】设置GPU与CPU的核绑(亲和力Affinity)

【教程】设置GPU与CPU的核绑(亲和力Affinity)

时间:2025/7/11 8:30:30来源:https://blog.csdn.net/sxf1061700625/article/details/139725011 浏览次数:0次

转载请注明出处:小锋学长生活大爆炸[xfxuezhagn.cn]

如果本文帮助到了你,欢迎[点赞、收藏、关注]哦~


        简单来说,核绑,或者叫亲和力,就是将某个GPU与指定CPU核心进行绑定,从而尽可能提高效率。

        推荐与进程优先级一起用:

【教程】Linux设置进程的优先级-CSDN博客文章浏览阅读420次,点赞17次,收藏7次。对于时间敏感的任务调整进程优先级很有必要https://blog.csdn.net/sxf1061700625/article/details/139721354


        可以看NVIDIA推荐GPU与哪些CPU绑定:

nvidia-smi topo -m


        Python中的设置代码参考:

def set_cpu_affinity(rank, num_cores_per_gpu=4, start_core_index=0):num_cores = psutil.cpu_count(logical=True)core_ids = list(range(num_cores))# 计算起始和结束核的索引start_core = start_core_index + rank * num_cores_per_gpuend_core = start_core + num_cores_per_gpu# 获取要绑定的CPU核列表cpu_affinity = core_ids[start_core:end_core]# 设置当前进程的CPU核绑定p = psutil.Process(os.getpid())p.cpu_affinity(cpu_affinity)print(f">> GPU {rank} is bound to CPU cores {cpu_affinity}")

        用法示例:

set_cpu_affinity(rank)

        效果演示:

htop

关键字:【教程】设置GPU与CPU的核绑(亲和力Affinity)

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

责任编辑: