詳解Pytorch里的pin_memory 和 non_blocking

前言 pin_memory 和 non_blocking的作用分別是什么?網上看了很多解釋，只是稀里糊塗的有個感覺，就是用了這玩意速度能變快，但是不知所以然，這篇文章希望能幫助你解惑，也給自己做個筆記，以備日后查閱。 1. pin_memory 1.1 什么是鎖頁內存（Pinned ...