Table of Contents

Shortcuts

Function torch_tensorrt::set_device¶

Defined in File torch_tensorrt.h

Function Documentation¶

TORCHTRT_API void torch_tensorrt::set_device(const int gpu_id)¶

Set gpu device id.

Parameters: gpu_id – Sets gpu id using cudaSetDevice

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources