PyTorch如何限制CPU線程數(shù)目
更新時(shí)間：2024年02月20日 10:30:48 作者：培之
這篇文章主要介紹了PyTorch如何限制CPU線程數(shù)目問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教
PyTorch限制CPU線程數(shù)目
終端命令 htop ：查看 CPU 利用率
限制 python 的cpu 利用率：
import os cpu_num = 4 # 這里設(shè)置成你想運(yùn)行的CPU個(gè)數(shù) 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 的資源，導(dǎo)致 cpu 占用率過(guò)高。
在讀數(shù)據(jù)的時(shí)候設(shè)置 num_workers = 0，也可能會(huì)導(dǎo)致 cpu 占用率過(guò)高。
PyTorch程序占用較高CPU解決
在使用服務(wù)器運(yùn)行PyTorch程序時(shí)，使用TOP命令查看CPU狀態(tài)，發(fā)現(xiàn)CPU占用率竟然達(dá)到2000%，已經(jīng)影響到其他人正常使用服務(wù)器。
解決辦法
在main程序中加以下代碼：
torch.set_num_threads(1)
設(shè)置最多使用1個(gè)CPU核。問(wèn)題解決~
總結(jié)
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
在 Python 中,將字符串拼接成字節(jié)有多種方式,主要取決于字符串的編碼和具體需求,這些方法涵蓋了從基本字符串拼接到底層字節(jié)處理的多種場(chǎng)景,本文通過(guò)代碼給大家介紹的非常詳細(xì),需要的朋友可以參考下2026-03-03
pymysql就是聯(lián)通mysql實(shí)現(xiàn)python窗口與mysql數(shù)據(jù)庫(kù)交互的庫(kù)，十分方便操作，這篇文章將詳細(xì)說(shuō)明如何掌握并使用該庫(kù)，感興趣的可以了解一下2023-02-02
Python采集股票數(shù)據(jù)并制作可視化柱狀圖
這篇文章主要介紹了如何利用Python采集股票數(shù)據(jù)并制作可視化柱狀圖，文中的示例代碼講解詳細(xì)，快來(lái)跟上小編一起動(dòng)手嘗試一下吧2022-04-04
這篇文章主要介紹了python5個(gè)頂級(jí)的異步框架，幫助大家更好的利用python進(jìn)行web開發(fā)，感興趣的朋友可以了解下2020-09-09
Python使用ffmpy將amr格式的音頻轉(zhuǎn)化為mp3格式的例子
今天小編就為大家分享一篇Python使用ffmpy將amr格式的音頻轉(zhuǎn)化為mp3格式的例子，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-08-08
在數(shù)據(jù)可視化領(lǐng)域,三維圖形能夠以更直觀的方式展示數(shù)據(jù)之間的復(fù)雜關(guān)系,本文將為大家詳細(xì)介紹如何使用Python繪制三維填充折線圖,需要的小伙伴可以了解下2025-07-07
Python學(xué)習(xí)小技巧之列表項(xiàng)的排序
這篇文章主要給大家介紹了Python學(xué)習(xí)小技巧之列表項(xiàng)排序的相關(guān)資料，文中介紹的非常詳細(xì)，需要的朋友們可以參借鑒，下面跟著小編一起來(lái)學(xué)習(xí)學(xué)習(xí)吧。2017-05-05
Python使用SQLAlchemy連接ClickHouse數(shù)據(jù)庫(kù)
ClickHouse是一個(gè)開源的列式數(shù)據(jù)庫(kù)管理系統(tǒng),以其高速的實(shí)時(shí)數(shù)據(jù)分析能力著稱,本文主要介紹了Python使用SQLAlchemy連接ClickHouse數(shù)據(jù)庫(kù),感興趣的可以了解一下2025-04-04
Python中Numpy包的安裝與使用方法簡(jiǎn)明教程
這篇文章主要介紹了Python中Numpy包的安裝與使用方法,結(jié)合簡(jiǎn)單實(shí)例形式分析了Python使用pip命令在線與離線whl包安裝,以及使用numpy打印隨機(jī)數(shù)矩陣的操作技巧,需要的朋友可以參考下2018-07-07
這篇文章主要介紹了利用Python讀取CSV文件并計(jì)算某一列的均值和方差,這里利用了csv模塊來(lái)對(duì)文件進(jìn)行處理，文章通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-12-12