Pytorch限制或增加CPU使用的核数方式
更新时间：2024年02月20日 10:41:05 作者：一穷二白到年薪百万
这篇文章主要介绍了Pytorch限制或增加CPU使用的核数方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
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的线程数量；
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
大家好，本篇文章主要讲的是Python处理excel与txt文件详解，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下，方便下次浏览2021-12-12
这篇文章主要介绍了Linux下使用python调用top命令获得CPU利用率,本文直接给出实现代码,需要的朋友可以参考下2015-03-03
所有以双下划线__包起来的方法,统称为Magic Method(魔术方法),它是一种的特殊方法,这篇文章主要给大家介绍了关于Python中魔术方法的定义及一些常用方法,需要的朋友可以参考下2024-02-02
面向对象设计（OOD）不会特别要求面向对象编程语言。事实上OOD可以由纯结构化语言来实现，比如C，但如果想要构造具备对象性质和特点的数据类型，就需要在程序上作更多的努力。当一门语言内建OO特性，OO编程开发就会更加方便高效。2014-10-10
在Python中,requests库是处理HTTP请求的标准工具之一,这篇文章主要为大家介绍了如何在requests中实现请求的自动重试,感兴趣的小伙伴可以跟随小编一起学习一下2023-11-11
今天小编就为大家分享一篇python列表切片和嵌套列表取值操作详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
Django使用详解:ORM 的反向查找(related_name)
今天小编就为大家分享一篇Django使用详解:ORM 的反向查找(related_name)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
这篇文章主要介绍了python编写扎金花小程序的实例代码,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-02-02
这篇文章主要介绍了python lambda表达式（匿名函数）写法解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-09-09
Python使用Selenium批量自动化获取并下载图片的方法
在现代的Web开发中,自动化测试和数据抓取已经成为不可或缺的一部分,Selenium作为一款强大的自动化测试工具,可以用于批量获取网页上的图片,所以本文给大家介绍了Python如何使用Selenium批量自动化获取并下载图片的方法2024-11-11
最新评论