Table of Contents
Shortcuts

函数 torch_tensorrt::set_device

函数文档

TORCHTRT_API void torch_tensorrt::set_device(const int gpu_id)

设置GPU设备ID。

Parameters

gpu_id – 使用 cudaSetDevice 设置 GPU ID

为了分析流量并优化您的体验，我们在本网站上使用cookies。通过点击或浏览，您同意我们使用cookies。作为本网站的当前维护者，Facebook的Cookies政策适用。了解更多信息，包括可用的控制选项：Cookies Policy。