Tensorflow - 是否可以手動確定圖形中的哪些張量轉到GPU和CPU？

Question

我在Keras中創建了一個使用卷積和LSTM層的網絡。

我讀過tensorflow不能在GPU處理好LSTM層：http://minimaxir.com/2017/07/cpu-or-gpu/

事實上，我的網絡得到了在GPU慢（希望它不是一個進一步的問題）。

我希望tensorflow將所有的卷積運算都扔到GPU上，但保留所有的LSTM操作到CPU中。那可能嗎？因此，考慮到我可以找到張量流圖並確定它的每個張量，我可以手動定義每個張量執行其操作的位置嗎？

Answer 1

實施之後this網絡，你可以（如果你使用Tensorflow）嘗試以下方法：在這個文件

with tf.device('/cpu:0'): 
    # model layers which should be run on CPU go here 

Answer 2

看：Distributed TensorFlow Specifying distributed devices in your model