Pytorch转TensorRT范例代码

最新推荐文章于 2024-08-19 09:34:01 发布

AI算法网奇

最新推荐文章于 2024-08-19 09:34:01 发布

阅读量3.4k

点赞数 1

分类专栏： pytorch知识宝典 onnx

本文链接：https://blog.csdn.net/jacke121/article/details/88683563

版权

pytorch知识宝典同时被 2 个专栏收录

499 篇文章 226 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

onnx

96 篇文章 25 订阅

订阅专栏

原文：https://blog.csdn.net/quantum7/article/details/83380935

Pytorch转TensorRT范例代码
　　TensorRT官方文档说，/usr/src/tensorrt/samples/python/network_api_pytorch_mnist下有示例代码。实际上根本就没有。这里提供一个示例代码，供参考。

　　这个范例的具体位置是：/usr/local/lib/python3.5/site-packages/tensorrt/examples/pytorch_to_trt



#!/usr/bin/python
import os
from random import randint
import numpy as np
 
try:
    import pycuda.driver as cuda
    import pycuda.gpuarray as gpuarray
    import pycuda.autoinit
except ImportError as err:
    raise ImportError("""ERROR: Failed to import module({})
Please make sure you have pycuda and the example dependencies i

了解本专栏