神经网络加速基础知识

最新推荐文章于 2024-07-25 22:24:14 发布

麻花地

最新推荐文章于 2024-07-25 22:24:14 发布

阅读量118

点赞数

分类专栏：深度学习环境深度学习使用模型文章标签：学习计算机视觉人工智能

本文链接：https://blog.csdn.net/charles_zhang_/article/details/128698147

版权

深度学习同时被 3 个专栏收录

74 篇文章 6 订阅

订阅专栏

深度学习环境

42 篇文章 3 订阅

订阅专栏

使用模型

10 篇文章 0 订阅

订阅专栏

神经网络加速基础知识

执行过程

指令执行时间

处理器

算力瓶颈计算代码

import torch
import torchvision
from tqdm import tqdm

DEVICE = "cuda:0"

model = torchvision.models.mobilenet_v2(pretrained=True)
model = model.to(DEVICE)

with torch.no_grad():
    data = torch.rand(size=[1,3,224,224])
    for i in tqdm(range(1024)):
        o = model.forward(data.to(DEVICE))

    data = torch.rand(size=[128,3,224,224])

    for i in tqdm(range(128)):
        o = model.forward(data.to(DEVICE))