How to enable Keras with Theano to utilize multiple GPUs

前端 未结 1 809
名媛妹妹
名媛妹妹 2021-02-08 13:57

Setup:

  • Using a Amazon Linux system with a Nvidia GPU
  • I\'m using Keras 1.0.1
  • Running Theano v0.8.2 backend
  • Using CUDA and CuDNN
相关标签:
1条回答
  • 2021-02-08 14:37

    Multi-GPU training is experimental ("The code is rather new and is still considered experimental at this point. It has been tested and seems to perform correctly in all cases observed, but make sure to double-check your results before publishing a paper or anything of the sort.") and hasn't been integrated into Keras yet. However, you can use multiple GPUs with Keras with the Tensorflow backend: https://blog.keras.io/keras-as-a-simplified-interface-to-tensorflow-tutorial.html#multi-gpu-and-distributed-training.

    0 讨论(0)
提交回复
热议问题