Memory 有没有什么方法可以“用”呢;交换;tensorflow中的GPU内存?

Memory 有没有什么方法可以“用”呢;交换;tensorflow中的GPU内存?,memory,tensorflow,Memory,Tensorflow,我正在训练一个超过GPU内存(超过11GB)的大型型号。我想知道tensorflow中是否有办法将GPU内存交换到主内存中。一些效率损失是可以接受的。完全在CPU上训练模型可以解决内存问题,但速度太慢。tensorflow中有一些类可以使用swap_内存参数创建网络 e、 g.对于RNN,您可以使用tf.nn.dynamic\u RNN或tf.nn.raw\u RNN 还有一个更通用的循环类tf.while\u loop,使用此参数 但是对于使用内存交换,没有通用的选项 只要看看tensorfl

我正在训练一个超过GPU内存(超过11GB)的大型型号。我想知道tensorflow中是否有办法将GPU内存交换到主内存中。一些效率损失是可以接受的。完全在CPU上训练模型可以解决内存问题,但速度太慢。

tensorflow中有一些类可以使用swap_内存参数创建网络

e、 g.对于RNN,您可以使用tf.nn.dynamic\u RNNtf.nn.raw\u RNN

还有一个更通用的循环类tf.while\u loop,使用此参数

但是对于使用内存交换,没有通用的选项

只要看看tensorflow.org并使用它的搜索功能。您可以使用swap_内存查找相关类