How to set the number of CPUs to be used

Hello. When using GraphST, I tried to set the number of CPUs to be used through torch.set_num_threads. However, I found that the setting didn't take effect when actually using it. So how should I set it? Thanks.