函数 torch_tensorrt::set_device¶
函数文档¶
-
TORCHTRT_API void torch_tensorrt::set_device(const int gpu_id)¶
设置 GPU 设备 ID。
- 参数
gpu_id – 使用 cudaSetDevice 设置 GPU ID
- 文档 >
- Torch-TensorRT C++ API >
- 函数 torch_tensorrt::set_device
-
快捷方式
