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