Tensorflow 限制keras中的CUDA核

我使用Keras和Tensorflow后端解决分类问题。可以限制我的程序使用的CUDA内核吗？我已经设法限制GPU内存使用以并行运行多个支持cuda的程序，但我还需要限制cuda内核