PyTorch如何限制CPU线程数目

更新时间：2024年02月20日 10:30:48 作者：培之

这篇文章主要介绍了PyTorch如何限制CPU线程数目问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

PyTorch限制CPU线程数目

终端命令 htop ：查看 CPU 利用率

限制 python 的cpu 利用率：

import os
cpu_num = 4 # 这里设置成你想运行的CPU个数
os.environ["OMP_NUM_THREADS"] = str(cpu_num)  # noqa
os.environ["MKL_NUM_THREADS"] = str(cpu_num) # noqa
torch.set_num_threads(cpu_num )

原因

tensor 没有 .to('cuda') ,tensor 还是部署在 cpu 上，耗的是 cpu 的资源，导致 cpu 占用率过高。

在读数据的时候设置 num_workers = 0，也可能会导致 cpu 占用率过高。

PyTorch程序占用较高CPU解决

在使用服务器运行PyTorch程序时，使用TOP命令查看CPU状态，发现CPU占用率竟然达到2000%，已经影响到其他人正常使用服务器。

解决办法

在main程序中加以下代码：

torch.set_num_threads(1)

设置最多使用1个CPU核。问题解决~

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python正则表达式匹配日期与时间的方法
这篇文章主要介绍了Python正则表达式匹配日期与时间的方法，本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下
2019-07-07
Python模块的制作方法实例分析
这篇文章主要介绍了Python模块的制作方法,结合实例形式分析了Python第三方模块的编写、调用相关操作技巧与使用注意事项,需要的朋友可以参考下
2019-12-12
Python3中的f-Strings增强版字符串格式化方法
这篇文章主要介绍了Python3中的f-Strings增强版字符串格式化方法,看完本文你将学习到如何以及为什么使用f-strings。对大家的工作或学习具有一定的参考借鉴价值，需要的朋友可以参考下
2020-03-03
Python的加密模块md5、sha、crypt使用实例
这篇文章主要介绍了Python的加密模块md5、sha、crypt使用实例,本文给出了MD5和crypt模块的代码实例,需要的朋友可以参考下
2014-09-09
详解Python函数print用法
今天给大家带来的是关于Python的相关知识,文章围绕着Python print函数的用法展开,文中有非常详细的介绍及代码示例,需要的朋友可以参考下
2021-06-06
Python+OpenCV读写视频的方法详解
这篇文章主要为大家详细介绍了Python+OpenCV进行读写视频操作的示例代码，文中的示例代码讲解详细，感兴趣的小伙伴可以动手尝试一下
2022-08-08
python可视化分析绘制带趋势线的散点图和边缘直方图
这篇文章主要介绍了python可视化分析绘制带趋势线的散点图和边缘直方图，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下
2022-06-06
python调用文字识别OCR轻松搞定验证码
本文主要介绍了python调用文字识别OCR轻松搞定验证码，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-11-11
如何使用pandas对超大csv文件进行快速拆分详解
Pandas是Python语言的一个扩展程序库,提供高性能、易于使用的数据结构和数据分析工具,下面这篇文章主要给大家介绍了关于如何使用pandas对超大csv文件进行快速拆分的相关资料,需要的朋友可以参考下
2022-07-07
详解Python图像形态学处理(开运算,闭运算,梯度运算)
数学形态学（Mathematical Morphology）是一种应用于图像处理和模式识别领域的新方法。本文将为大家介绍Python图像形态学处理中的开运算、闭运算和梯度运算，感兴趣的可以了解一下
2022-06-06