From 3e87247d83bd4d7184c3ec11a1145b659bc5c718 Mon Sep 17 00:00:00 2001 From: Dw9 Date: Fri, 25 Jul 2025 16:09:23 +0800 Subject: [PATCH] update networks.py --- clearvoice/clearvoice/networks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clearvoice/clearvoice/networks.py b/clearvoice/clearvoice/networks.py index e46c125..93068fc 100644 --- a/clearvoice/clearvoice/networks.py +++ b/clearvoice/clearvoice/networks.py @@ -63,7 +63,7 @@ def __init__(self, args): if free_gpu_id is not None: args.use_cuda = 1 torch.cuda.set_device(free_gpu_id) - self.device = torch.device('cuda') + self.device = torch.device(f'cuda:{free_gpu_id}') else: # If no GPU is detected, use the CPU #print("No GPU found. Using CPU.")