PyTorch如何限制CPU線程數(shù)目
更新時間：2024年02月20日 10:30:48 作者：培之
這篇文章主要介紹了PyTorch如何限制CPU線程數(shù)目問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
PyTorch限制CPU線程數(shù)目
終端命令 htop ：查看 CPU 利用率
限制 python 的cpu 利用率：
import os cpu_num = 4 # 這里設置成你想運行的CPU個數(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 的資源，導致 cpu 占用率過高。
在讀數(shù)據(jù)的時候設置 num_workers = 0，也可能會導致 cpu 占用率過高。
PyTorch程序占用較高CPU解決
在使用服務器運行PyTorch程序時，使用TOP命令查看CPU狀態(tài)，發(fā)現(xiàn)CPU占用率竟然達到2000%，已經(jīng)影響到其他人正常使用服務器。
解決辦法
在main程序中加以下代碼：
torch.set_num_threads(1)
設置最多使用1個CPU核。問題解決~
總結(jié)
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了Python正則表達式匹配日期與時間的方法，本文通過實例代碼給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了Python模塊的制作方法,結(jié)合實例形式分析了Python第三方模塊的編寫、調(diào)用相關(guān)操作技巧與使用注意事項,需要的朋友可以參考下2019-12-12
這篇文章主要介紹了Python3中的f-Strings增強版字符串格式化方法,看完本文你將學習到如何以及為什么使用f-strings。對大家的工作或?qū)W習具有一定的參考借鑒價值，需要的朋友可以參考下2020-03-03
這篇文章主要介紹了Python的加密模塊md5、sha、crypt使用實例,本文給出了MD5和crypt模塊的代碼實例,需要的朋友可以參考下2014-09-09
今天給大家?guī)淼氖顷P(guān)于Python的相關(guān)知識,文章圍繞著Python print函數(shù)的用法展開,文中有非常詳細的介紹及代碼示例,需要的朋友可以參考下2021-06-06
這篇文章主要為大家詳細介紹了Python+OpenCV進行讀寫視頻操作的示例代碼，文中的示例代碼講解詳細，感興趣的小伙伴可以動手嘗試一下2022-08-08
這篇文章主要介紹了python可視化分析繪制帶趨勢線的散點圖和邊緣直方圖，文章圍繞主題展開詳細的內(nèi)容介紹，具有一定的參考價值，需要的小伙伴可以參考一下2022-06-06
本文主要介紹了python調(diào)用文字識別OCR輕松搞定驗證碼，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-11-11
Pandas是Python語言的一個擴展程序庫,提供高性能、易于使用的數(shù)據(jù)結(jié)構(gòu)和數(shù)據(jù)分析工具,下面這篇文章主要給大家介紹了關(guān)于如何使用pandas對超大csv文件進行快速拆分的相關(guān)資料,需要的朋友可以參考下2022-07-07
詳解Python圖像形態(tài)學處理(開運算,閉運算,梯度運算)
數(shù)學形態(tài)學（Mathematical Morphology）是一種應用于圖像處理和模式識別領域的新方法。本文將為大家介紹Python圖像形態(tài)學處理中的開運算、閉運算和梯度運算，感興趣的可以了解一下2022-06-06