OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
71cb13869b
pytorch/caffe2/contrib/tensorrt
History
Kazuaki Ishizaki a5f04e9a91 Fix typos in .md and .rst files (#88962) 
This PR fixes typos `Github` in `.md` and `.rst` files.
`Github` -> `GitHub`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/88962
Approved by: https://github.com/kit1980
2022-11-17 03:37:02 +00:00
..
CMakeLists.txt
README.md
tensorrt_op_trt.cc
tensorrt_op_trt.h
tensorrt_tranformer.cc
tensorrt_tranformer.h
trt_utils.cc
trt_utils.h

README.md

Caffe2 & TensorRT integration

Jenkins Build Status

This directory contains the code implementing TensorRTOp Caffe2 operator as well as Caffe2 model converter (using ONNX model as an intermediate format). To enable this functionality in your PyTorch build please set

USE_TENSORRT=1 ... python setup.py ...

or if you use CMake directly

-DUSE_TENSORRT=ON

For further information please explore caffe2/python/trt/test_trt.py test showing all possible use cases.

Questions and Feedback

Please use GitHub issues (https://github.com/pytorch/pytorch/issues) to ask questions, report bugs, and request new features.