CUDAの使い方tensorflow pythonライブラリの固定メモリ

Question

現在、GPUでテンソルフローを実行しています。私はCUDA固定メモリについて読んでいました。

テンソルフローのPythonライブラリを使用しているときに、これを設定することができませんでした。 どうすればいいですか？

Answer 1

CPUとGPUの間のmemcpyに自動的に使用されます。より洗練された機能が必要な場合は、明示的に使用するカーネルを書くことができます。