Skip to content

Fix 'torch.device' object has no attribute 'gpu_id' issue in trt export - #8019

Merged
KumoLiu merged 5 commits into
Project-MONAI:devfrom
KumoLiu:fix-trt
Aug 15, 2024
Merged

Fix 'torch.device' object has no attribute 'gpu_id' issue in trt export#8019
KumoLiu merged 5 commits into
Project-MONAI:devfrom
KumoLiu:fix-trt

use cuda set device

5e21a00
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs