CPU control

[TheSallyGardens]

Hello, I've try your SCALE tools.I want to know how you can control the CPU usage.

[jsxlei]

SCALE can only use single core of CPU for now while supports multi-GPUs on cuda device.

[TheSallyGardens]

SCALE can only use single core of CPU for now while supports multi-GPUs on cuda device. I have 56 cores on my server.When I use SCALE，It takes up the whole core.

[TheSallyGardens]

SCALE can only use single core of CPU for now while supports multi-GPUs on cuda device.

For training data.

[jsxlei]

Try torch.set_num_threads(1) or refer to this link: https://github.com/pytorch/pytorch/issues/16894.

[jsxlei]

Add OMP_NUM_THREADS=1 or MKL_NUM_THREADS=1 before SCALE.py. e.g. OMP_NUM_THREADS=1 SCALE.py -d xx which works on our centos system, to use multi-cores, just modify the OMP_NUM_THREADS=num you want

[TheSallyGardens]

Add OMP_NUM_THREADS=1 or MKL_NUM_THREADS=1 before SCALE.py. e.g. OMP_NUM_THREADS=1 SCALE.py -d xx which works on our centos system, to use multi-cores, just modify the OMP_NUM_THREADS=num you want

Thank you for your answer.