Function trtorch::set_device

Function Documentation

TRTORCH_API void trtorch :: set_device ( const int gpu_id )

Set gpu device id.

Parameters
  • gpu_id : Sets gpu id using cudaSetDevice