函式 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