torch.set_num_threads

 

不设置，cpu51%，时间15ms

2的时候，cpu17%,时间15ms变为25ms

4的时候，cpu34%，时间17ms

8的时候，cpu67%，

import time

import torch
import numpy as np

while True:
    torch.set_num_threads(4)
    arr= np.zeros((100,3, 300, 300), dtype=np.float32)

    # start=time.time()
    # arr=arr-0.406
    # print('time1',time.time()-start)
    aaa=torch.from_numpy(arr)
    start = time.time()
    aaa.add_(-0.406)
    print('time2',time.time() - start)
    # aaa.add_(-0.406)
    # print(aaa)
    # inps = torch.zeros(50, 3, 300, 300)
    time.sleep(0.007)