How to control the CPU kernels that are used during training.

[Juicechen95]

I am running this code on a server. I use CUDA_VISIBLE_DEVICES=0 to ensure that I only use 1 GPU. But this code uses nearly all the CPU kernels during the training time, which make other users difficult to use the server. I want to know how to control the CPU usage?? Thanks a lot.