2021年7月26日
torch之DataLoader参数pin_memory解析
摘要： 关于什么是锁页内存： pin_memory就是锁页内存，创建DataLoader时，设置pin_memory=True，则意味着生成的Tensor数据最开始是属于内存中锁页内存，这样将内存的Tensor转义到GPU的显存就会更快一些。 主机中的内存，有两种存在方法，一是锁页，二是不索页，锁页内存存放 阅读全文
posted @ 2021-07-26 15:23 重大的小鸿 阅读(1654) 评论(0) 推荐(1)