Function trtorch::set_device ¶
-
Defined in File trtorch.h
Function Documentation ¶
-
TRTORCH_API void
trtorch
::
set_device
( const int gpu_id ) ¶
-
Set gpu device id.
- Parameters
-
-
gpu_id
: Sets gpu id using cudaSetDevice
-