Keras Tf2多处理cuda线程

我正在做一些强化学习，并试图在我的CPU核心多线程不同的学习者

我正在使用带有mpi进程的python上的tf2 gpu enable

问题是，每个线程都在尝试加载cuda并失败，然后崩溃

我想有我的主循环做梯度gpu启用和线程不加载cuda

如果有人能指引我，那将是令人惊奇的