详解Pytorch里的pin_memory 和 non_blocking

前言 pin_memory 和 non_blocking的作用分别是什么?网上看了很多解释，只是稀里糊涂的有个感觉，就是用了这玩意速度能变快，但是不知所以然，这篇文章希望能帮助你解惑，也给自己做个笔记，以备日后查阅。 1. pin_memory 1.1 什么是锁页内存（Pinned ...