pytorch創(chuàng)建data.DataLoader時(shí)對(duì)參數(shù)pin_memory的理解是什么

這篇文章將為大家詳細(xì)講解有關(guān)pytorch創(chuàng)建data.DataLoader時(shí)對(duì)參數(shù)pin_memory的理解是什么,文章內(nèi)容質(zhì)量較高,因此小編分享給大家做個(gè)參考,希望大家閱讀完這篇文章后對(duì)相關(guān)知識(shí)有一定的了解。

創(chuàng)新互聯(lián)公司憑借在網(wǎng)站建設(shè)、網(wǎng)站推廣領(lǐng)域領(lǐng)先的技術(shù)能力和多年的行業(yè)經(jīng)驗(yàn),為客戶提供超值的營(yíng)銷型網(wǎng)站建設(shè)服務(wù),我們始終認(rèn)為:好的營(yíng)銷型網(wǎng)站就是好的業(yè)務(wù)員。我們已成功為企業(yè)單位、個(gè)人等客戶提供了網(wǎng)站設(shè)計(jì)、成都做網(wǎng)站服務(wù),以良好的商業(yè)信譽(yù),完善的服務(wù)及深厚的技術(shù)力量處于同行領(lǐng)先地位。

pytorch創(chuàng)建data.DataLoader時(shí),參數(shù)pin_memory的理解

pin_memory就是鎖頁(yè)內(nèi)存,創(chuàng)建DataLoader時(shí),設(shè)置pin_memory=True,
則意味著生成的Tensor數(shù)據(jù)最開始是屬于內(nèi)存中的鎖頁(yè)內(nèi)存,
這樣將內(nèi)存的Tensor轉(zhuǎn)義到GPU的顯存就會(huì)更快一些。

主機(jī)中的內(nèi)存,有兩種存在方式,一是鎖頁(yè),二是不鎖頁(yè),
鎖頁(yè)內(nèi)存存放的內(nèi)容在任何情況下都不會(huì)與主機(jī)的虛擬內(nèi)存
進(jìn)行交換(注:虛擬內(nèi)存就是硬盤),而不鎖頁(yè)內(nèi)存在主機(jī)內(nèi)存不足時(shí),
數(shù)據(jù)會(huì)存放在虛擬內(nèi)存中。

而顯卡中的顯存全部是鎖頁(yè)內(nèi)存!

當(dāng)計(jì)算機(jī)的內(nèi)存充足的時(shí)候,可以設(shè)置pin_memory=True。
當(dāng)系統(tǒng)卡住,或者交換內(nèi)存使用過多的時(shí)候,設(shè)置pin_memory=False。
因?yàn)閜in_memory與電腦硬件性能有關(guān),pytorch開發(fā)者不能確保每
一個(gè)煉丹玩家都有高端設(shè)備,因此pin_memory默認(rèn)為False。

關(guān)于pytorch創(chuàng)建data.DataLoader時(shí)對(duì)參數(shù)pin_memory的理解是什么就分享到這里了,希望以上內(nèi)容可以對(duì)大家有一定的幫助,可以學(xué)到更多知識(shí)。如果覺得文章不錯(cuò),可以把它分享出去讓更多的人看到。

新聞標(biāo)題:pytorch創(chuàng)建data.DataLoader時(shí)對(duì)參數(shù)pin_memory的理解是什么
本文路徑:http://muchs.cn/article10/igspgo.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供靜態(tài)網(wǎng)站、移動(dòng)網(wǎng)站建設(shè)、定制開發(fā)、動(dòng)態(tài)網(wǎng)站、商城網(wǎng)站域名注冊(cè)

廣告

聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請(qǐng)盡快告知,我們將會(huì)在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如需處理請(qǐng)聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時(shí)需注明來(lái)源: 創(chuàng)新互聯(lián)

商城網(wǎng)站建設(shè)