标签: numba numba-pro
在Pytorch中,有一个选项可以固定CPU阵列以快速传输到GPU(尽管似乎不适用于GPU-> CPU)。
我想知道是否有一种方法可以将Numba阵列固定到内存,或者有其他替代技术可以将CPU快速传输到GPU。我在文档中没有看到直接的选项,因此我想我们需要在数组创建期间指定分配。