Pytorch限制或增加CPU使用的核数方式
更新时间：2024年02月20日 10:41:05 作者：一穷二白到年薪百万
这篇文章主要介绍了Pytorch限制或增加CPU使用的核数方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
（福利推荐：【腾讯云】服务器最新限时优惠活动，云服务器1核2G仅99元/年、2核4G仅768元/3年，立即抢购>>>：9i0i.cn/qcloud）
（福利推荐：你还在原价购买阿里云服务器？现在阿里云0.8折限时抢购活动来啦！4核8G企业云服务器仅2998元/3年，立即抢购>>>：9i0i.cn/aliyun）
Pytorch限制或增加CPU的核数
一般Pytorch 默认是使用一半的CPU运行的，有的时候用不到那么多得CPU资源或者占用太多资源挤压别人，这就需要限制核数。
Pytorch提供了一个限制核数的代码：
# cpu_num是一个整数 torch.set_num_threads(cpu_num)
这样就能够限制核数。
Python获取CPU的线程数量
在使用PyTorch搭建目标检测模型时，我们需要设置最佳的数据读取的进程数量；
我觉得，这个数量最好就设置成CPU的线程数量；
获取CPU的线程数量
获取CPU线程数量的代码：
from multiprocessing import cpu_count print(cpu_count())
备注：
Python官方文档中关于multiprocessing.cpu_count()的功能是这样说的：
Return the number of CPUs in the system.
我先以为是文档描述的不准确，后来
Andrew Svetlov在我的Python文档Bug提交中告诉我：
CPU as a shortcut for "logical processor" is very common in software engineering.
所以这里的“the number of CPUs”实际上就是指的系统可以获取到的CPU逻辑核心数；
如果是对于单个CPU而言，那就是当前CPU的线程数量；
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持程序员之家。
相关文章
logging是Python标准库中记录常用的记录日志库，主要用于输出运行日志，可以设置输出日志的等级、日志保存路径、日志文件回滚等。本文主要来和大家聊聊它的具体用法，希望对大家有所帮助2023-02-02
这篇文章主要介绍了Python标准库json模块和pickle模块使用详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-03-03
DataFrame 数据合并实现(merge，join，concat)
这篇文章主要介绍了DataFrame 数据合并实现(merge，join，concat)，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-06-06
下面小编就为大家带来一篇python魔法方法-自定义序列详解。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2016-07-07
这篇文章主要介绍了python3.9之你应该知道的新特性详解,需要的朋友可以参考下2021-04-04
这篇文章主要为大家详细介绍了如何用Python合并lmdb文件，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-07-07
这篇文章主要为大家详细介绍了简洁版的Python学生成绩管理系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-01-01
本文主要通过源码分析，总结出一套简洁的requests请求类封装，帮助大家更好的由浅入深的理解python接口自动化，希望对大家的python接口自动化学习有所帮助2021-08-08
这篇文章主要介绍了Python异常模块traceback用法,结合实例形式分析了Python异常模块traceback的基本功能、使用方法及相关操作注意事项,需要的朋友可以参考下2019-10-10
这篇文章主要介绍了关于python3中setup.py小概念解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
最新评论