diff --git a/auto3dseg/algorithm_templates/dints/scripts/dummy_runner.py b/auto3dseg/algorithm_templates/dints/scripts/dummy_runner.py
index 675dc36c..daf5f0a7 100644
--- a/auto3dseg/algorithm_templates/dints/scripts/dummy_runner.py
+++ b/auto3dseg/algorithm_templates/dints/scripts/dummy_runner.py
@@ -38,7 +38,6 @@ def __init__(self, output_path, data_stats_file, device_id: int = 0):
         parser.read_config(config_file)
 
         self.device = torch.device("cuda:{0:d}".format(device_id))
-        torch.cuda.set_device(self.device)
 
         self.input_channels = parser.get_parsed_content("training#input_channels")
         self.patch_size = parser.get_parsed_content("training#patch_size")
diff --git a/auto3dseg/algorithm_templates/segresnet2d/scripts/dummy_runner.py b/auto3dseg/algorithm_templates/segresnet2d/scripts/dummy_runner.py
index 4767af09..96461ec7 100644
--- a/auto3dseg/algorithm_templates/segresnet2d/scripts/dummy_runner.py
+++ b/auto3dseg/algorithm_templates/segresnet2d/scripts/dummy_runner.py
@@ -38,7 +38,6 @@ def __init__(self, output_path, data_stats_file, device_id: int = 0):
         parser.read_config(config_file)
 
         self.device = torch.device("cuda:{0:d}".format(device_id))
-        torch.cuda.set_device(self.device)
 
         self.input_channels = parser.get_parsed_content("input_channels")
         self.num_adjacent_slices = parser.get_parsed_content("num_adjacent_slices")
diff --git a/auto3dseg/algorithm_templates/swinunetr/scripts/dummy_runner.py b/auto3dseg/algorithm_templates/swinunetr/scripts/dummy_runner.py
index 1bfb5200..d7480093 100644
--- a/auto3dseg/algorithm_templates/swinunetr/scripts/dummy_runner.py
+++ b/auto3dseg/algorithm_templates/swinunetr/scripts/dummy_runner.py
@@ -38,7 +38,6 @@ def __init__(self, output_path, data_stats_file, device_id: int = 0):
         parser.read_config(config_file)
 
         self.device = torch.device("cuda:{0:d}".format(device_id))
-        torch.cuda.set_device(self.device)
 
         self.input_channels = parser.get_parsed_content("input_channels")
         self.patch_size = parser.get_parsed_content("patch_size")