Keras强制数据集位于VRAM中,或使用多个CPU内核将数据从RAM馈送到GPU

时间:2018-12-16 17:12:26

标签: tensorflow keras

我在TensorFlow后端上使用Keras,并注意到我的GPU不能足够快地获取数据。
我正在使用的数据集本来是Pandas DataFrame,但经过预处理后,它由4个NumPy数组(X_train,y_train,X_test,y_test)组成,很容易放入RAM。 如何创建使用多个CPU内核的管道,以将数据从RAM发送到GPU内存?

所有数据已经​​在RAM中,但也应轻松放入GPU内存中。
也可以选择强制使用它,但是创建管道对于一般用途来说更有趣。

谢谢。

0 个答案:

没有答案