如何使用CUDA從tensorflow python庫中固定內存

Question

我目前正在研究GPU上的運行張量流。我正在閱讀關於CUDA固定內存。

我無法找到使用tensorflow python庫設置此。 任何想法如何完成？

Answer 1

它自動用於CPU和GPU之間的任何memcpy。如果你想要更復雜的功能，你可以編寫一個明確使用它們的內核。