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 中,将字符串拼接成字节有多种方式,主要取决于字符串的编码和具体需求,这些方法涵盖了从基本字符串拼接到底层字节处理的多种场景,本文通过代码给大家介绍的非常详细,需要的朋友可以参考下2026-03-03
pymysql就是联通mysql实现python窗口与mysql数据库交互的库，十分方便操作，这篇文章将详细说明如何掌握并使用该库，感兴趣的可以了解一下2023-02-02
这篇文章主要介绍了如何利用Python采集股票数据并制作可视化柱状图，文中的示例代码讲解详细，快来跟上小编一起动手尝试一下吧2022-04-04
这篇文章主要介绍了python5个顶级的异步框架，帮助大家更好的利用python进行web开发，感兴趣的朋友可以了解下2020-09-09
Python使用ffmpy将amr格式的音频转化为mp3格式的例子
今天小编就为大家分享一篇Python使用ffmpy将amr格式的音频转化为mp3格式的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
在数据可视化领域,三维图形能够以更直观的方式展示数据之间的复杂关系,本文将为大家详细介绍如何使用Python绘制三维填充折线图,需要的小伙伴可以了解下2025-07-07
这篇文章主要给大家介绍了Python学习小技巧之列表项排序的相关资料，文中介绍的非常详细，需要的朋友们可以参借鉴，下面跟着小编一起来学习学习吧。2017-05-05
Python使用SQLAlchemy连接ClickHouse数据库
ClickHouse是一个开源的列式数据库管理系统,以其高速的实时数据分析能力著称,本文主要介绍了Python使用SQLAlchemy连接ClickHouse数据库,感兴趣的可以了解一下2025-04-04
这篇文章主要介绍了Python中Numpy包的安装与使用方法,结合简单实例形式分析了Python使用pip命令在线与离线whl包安装,以及使用numpy打印随机数矩阵的操作技巧,需要的朋友可以参考下2018-07-07
这篇文章主要介绍了利用Python读取CSV文件并计算某一列的均值和方差,这里利用了csv模块来对文件进行处理，文章通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-12-12
最新评论