torch div优化

最新推荐文章于 2024-03-27 22:50:45 发布

AI算法网奇

最新推荐文章于 2024-03-27 22:50:45 发布

阅读量3.5k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch知识宝典

本文链接：https://blog.youkuaiyun.com/jacke121/article/details/84262114

pytorch知识宝典专栏收录该内容

512 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文探讨了在PyTorch中进行除法运算的不同方法，包括CPU和CUDA的性能比较。通过实验发现，使用CUDA执行除法操作可以显著提高速度，达到2.5秒，相比Python原生实现的11.5秒快了约4.6倍，而在CPU上，PyTorch相比于Python也展现出近一倍的速度优势。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

下面的方法比上面的快2倍左右，cpu占用差不多。

torch1.0好像没有这个优势了

python 11.5s

torch能快一倍左右，6s

import cv2
if __name__ == '__main__':
    path=r'D:\data\Original\JPEGImages\09981.jpg'

    img_ = cv2.imread(path)
    start = time.time()
    for j in range(500):
        #python中如果是除法，img=img/255.0 会很快;img1=img/255.0 就会比较慢，一张图片需要156ms
        img=img_/255.0
        img2 = torch.from_numpy(img).float().unsqueeze(0)
        # time.sleep(0.007)
    print('time1',time.time()-start)
    img = cv2.imread(path)
    start = time.time()
    for j in range(500):
        # i

了解本专栏