限制keras中的CUDA核心

Question

我正在使用带有Tensorflow后端的Keras来解决分类问题。是否可以限制我的程序使用的CUDA内核？我设法限制了GPU内存的使用，以并行运行多个启用了cuda的程序，但我还需要限制CUDA内核。