zoukankan      html  css  js  c++  java
  • torch之DataLoader参数pin_memory解析

    关于什么是锁页内存:

    pin_memory就是锁页内存,创建DataLoader时,设置pin_memory=True,则意味着生成的Tensor数据最开始是属于内存中锁页内存,这样将内存的Tensor转义到GPU的显存就会更快一些。

    主机中的内存,有两种存在方法,一是锁页,二是不索页,锁页内存存放的内容在任何情况下都不会与主机的虚拟内存进行交换(注:虚拟内存就是硬盘),而不锁页内存在主机内存不足时,数据会存放在虚拟内存中。显卡中的显存全部是锁页内存,当计算机的内存充足的时候,可以设置pin_memory=True。当系统卡主,或者内存使用过多的时候,设置pin_memory=False。因为pin_memory与电脑硬件性能有关,pytorch开发者不能确定每一个炼丹玩家都有高端设备,因此pin_memory默认设置为False.

    原文链接:https://blog.csdn.net/yangwangnndd/article/details/95385628

  • 相关阅读:
    vue-实践1
    node 基本使用
    vue通信
    初始AMD和Common.js
    vue正确引入第三方包
    常见的java设计模式
    springboot加ES实现全局检索
    Cookie丢失的原因
    动态SQL
    用Java实现给图片添加文字
  • 原文地址:https://www.cnblogs.com/elitphil/p/15061625.html
Copyright © 2011-2022 走看看