Pytorch限制或增加CPU使用的核數(shù)方式
Pytorch限制或增加CPU的核數(shù)
一般Pytorch 默認(rèn)是使用一半的CPU運(yùn)行的，有的時(shí)候用不到那么多得CPU資源或者占用太多資源擠壓別人，這就需要限制核數(shù)。
Pytorch提供了一個(gè)限制核數(shù)的代碼：
# cpu_num是一個(gè)整數(shù) torch.set_num_threads(cpu_num)
這樣就能夠限制核數(shù)。
Python獲取CPU的線程數(shù)量
在使用PyTorch搭建目標(biāo)檢測(cè)模型時(shí)，我們需要設(shè)置最佳的數(shù)據(jù)讀取的進(jìn)程數(shù)量；
我覺得，這個(gè)數(shù)量最好就設(shè)置成CPU的線程數(shù)量；
獲取CPU的線程數(shù)量
獲取CPU線程數(shù)量的代碼：
from multiprocessing import cpu_count print(cpu_count())
備注：
Python官方文檔中關(guān)于multiprocessing.cpu_count()的功能是這樣說的：
Return the number of CPUs in the system.
我先以為是文檔描述的不準(zhǔn)確，后來
Andrew Svetlov在我的Python文檔Bug提交中告訴我：
CPU as a shortcut for "logical processor" is very common in software engineering.
所以這里的“the number of CPUs”實(shí)際上就是指的系統(tǒng)可以獲取到的CPU邏輯核心數(shù)；
如果是對(duì)于單個(gè)CPU而言，那就是當(dāng)前CPU的線程數(shù)量；
總結(jié)
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python中順序結(jié)構(gòu)和循環(huán)結(jié)構(gòu)示例代碼
這篇文章主要介紹了Python中的條件語(yǔ)句和循環(huán)語(yǔ)句,條件語(yǔ)句用于根據(jù)條件執(zhí)行不同的代碼塊,循環(huán)語(yǔ)句用于重復(fù)執(zhí)行一段代碼,文章還詳細(xì)說明了range函數(shù)的使用和continue與break語(yǔ)句的作用,需要的朋友可以參考下2025-02-02
深入理解Python中的 __new__ 和 __init__及區(qū)別介紹
這篇文章主要介紹了深入理解Python中的 __new__ 和 __init__及區(qū)別介紹,這兩個(gè)方法的主要區(qū)別在于：__new__ 負(fù)責(zé)對(duì)象的創(chuàng)建而 __init__ 負(fù)責(zé)對(duì)象的初始化。具體內(nèi)容詳情大家跟隨小編一起看看吧2018-09-09
Python雖然也提供了array模塊,但其只支持一維數(shù)組,不支持多維數(shù)組,也沒有各種運(yùn)算函數(shù),因而不適合數(shù)值運(yùn)算,NumPy的出現(xiàn)彌補(bǔ)了這些不足,這篇文章主要給大家介紹了關(guān)于python中NumPy的安裝與基本操作的相關(guān)資料,需要的朋友可以參考下2022-03-03
re.findall()?函數(shù)是?python?中正則表達(dá)式模塊（re）的一個(gè)重要函數(shù),它可以根據(jù)正則表達(dá)式搜索字符串,并返回匹配的字符串列表,這篇文章給大家介紹了python中re.findall()?的使用案例,感興趣的朋友跟隨小編一起看看吧2023-09-09
相信python的開發(fā)者對(duì)于python的鴨子類型比較熟悉，鴨子類型在維基百科中的準(zhǔn)確定義是‘是動(dòng)態(tài)類型的一種風(fēng)格。在這種風(fēng)格中，一個(gè)對(duì)象有效的語(yǔ)義，不是由繼承自特定的類或?qū)崿F(xiàn)特定的接口，而是由"當(dāng)前方法和屬性的集合"決定’。所以這篇文章給大家python的鴨子類型。2016-09-09
這篇文章主要介紹了Python引用傳值概念與用法,簡(jiǎn)單分析了Python引用傳值的概念、功能并結(jié)合實(shí)例形式總結(jié)分析了Python引用傳值的具體實(shí)現(xiàn)與使用方法,需要的朋友可以參考下2017-10-10
這篇文章將詳細(xì)講解圖像融合處理和ROI區(qū)域繪制，同時(shí)補(bǔ)充圖像屬性、通道和類型轉(zhuǎn)換。文中的示例代碼講解詳細(xì)，需要的可以參考一下2022-01-01
Python列表pop()函數(shù)使用實(shí)例詳解
這篇文章主要介紹了Python列表pop()函數(shù)使用實(shí)例詳解，本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2023-07-07
Python中使用pypdf2合并、分割、加密pdf文件的代碼詳解
這篇文章主要介紹了Python中使用pypdf2合并、分割、加密pdf文件的代碼，本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-05-05
Python關(guān)于OS文件目錄處理的實(shí)例分享
在本篇文章里小編給大家整理的是一篇關(guān)于Python關(guān)于OS文件目錄處理的實(shí)例內(nèi)容，有興趣的朋友們可以學(xué)一下。2021-05-05