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è)參考，也希望大家多多支持腳本之家。