主机中的内存,有两种存在方式,一是
锁页
,二是不锁页
,锁页内存存放的内容在任何情况下都不会与主机的虚拟内存进行交换(注:虚拟内存就是硬盘),而不锁页内存在主机内存不足时,数据会存放在虚拟内存中。
显卡中的显存全部是锁页内存。
创建DataLoader时,设置pin_memory=True,则意味着生成的Tensor数据最开始是属于内存中的锁页内存,这样将内存的Tensor转义到GPU的显存就会更快一些。
参考链接:
https://blog.csdn.net/tsq292978891/article/details/80454568
https://blog.csdn.net/shwan_ma/article/details/98741142