Function torch_tensorrt::set_device

• 定义于 文件 torch_tensorrt.h

函数文档

TORCHTRT_API void torch_tensorrt::set_device(const int gpu_id)

设置 GPU 设备 ID。

参数

gpu_id – 使用 cudaSetDevice 设置 GPU ID