Fixing Multiple GPU example limited to one GPU

Fixes #2563 Modifying data_parallel_tutorial.py, removed cuda:0 to enable using multiple GPUs.
pytorch · Nov 4, 2023 · c9f86f6 · c9f86f6
1 parent e7563f6
commit c9f86f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/beginner_source/blitz/data_parallel_tutorial.py b/beginner_source/blitz/data_parallel_tutorial.py
@@ -57,7 +57,7 @@
 ######################################################################
 # Device
 #
-device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 ######################################################################
 # Dummy DataSet