pytorch在cpu和gpu运算的性能差别

公共：

import time
import torch

print(torch.__version__)
print("torch.cuda.is_available()   =", torch.cuda.is_available())
print("torch.cuda.device_count()   =", torch.cuda.device_count())
print("torch.cuda.device('cuda')   =", torch.cuda.device('cuda'))
print("torch.cuda.current_device() =", torch.cuda.current_device())

cpu:7700k

start=time.time()
for i in range(1,10):
    a = torch.FloatTensor(1*100,1000,1000)
    a=torch.matmul(a,a)
    end=time.time()
    print (end-start)

gpu:1050ti

start=time.time()
for i in range(1,10):
    a = torch.FloatTensor(1*100,1000,1000)
    a = a.cuda()
    a=torch.matmul(a,a)
    end=time.time()
    print (end-start)

测试结果：

以下是租用服务器结果：（8255c 对比 2080ti）