如何使用来自tensorflow python库的CUDA固定内存

Question

我目前正在考虑在GPU上运行tensorflow。我正在阅读有关CUDA固定内存的信息。

在使用tensorflow python库时，我无法找到设置它。 知道如何做到这一点吗？

Answer 1

它自动用于CPU和GPU之间的任何memcpy。如果您需要更复杂的功能，可以编写一个明确使用它们的内核。