【triton-inference-server】 官方python_backend 的AddSubNet PyTorch 例子分析

阅读 21

2024-07-24

AddSubNet PyTorch example.

在这里插入图片描述

AddSubNet in PyTorch

In order to use this model, you need to install PyTorch. We recommend using pip method mentioned in the PyTorch website. Make sure that PyTorch is available in the same Python environment as other dependencies. Alternatively, you can create a Python Execution Environment. You can find the files for this example in examples/pytorch.
在这里插入图片描述

  • 模型:
  • https://github.com/triton-inference-server/python_backend/blob/main/examples/pytorch/model.py
# Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.

精彩评论(0)

0 0 举报