cpu比gpu快的情况

# -*- coding: utf-8 -*-
import torch
import time
from tqdm import tqdm

device=torch.device("cuda" if torch.cuda.is_available() else "cpu")

# CPU
start_time=time.time()
a=torch.ones(4,4)
for _ in tqdm(range(int(1e7))):
    a+=a
elapsed_time=time.time()-start_time
print('cpu time=',elapsed_time)

#GPU
start_time=time.time()
b=torch.ones(4,4).to(device)
for _ in tqdm(range(int(1e7))):
    b+=b
elapsed_time=time.time()-start_time
print('GPU time=',elapsed_time)

输出为：

cpu time= 17.210890293121338
GPU time=

思考： 在不修改程序结构的情况下，只修改其中部分数字，怎样让gpu计算的比cpu更快呢？

留言区评论吧！