Update data_parallel_tutorial.py to remove cuda:0

Removing cuda:0 from the comments
pytorch · Nov 5, 2023 · 2821752 · 2821752
1 parent c9f86f6
commit 2821752
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
@@ -9,7 +9,7 @@
 
 .. code:: python
 
-    device = torch.device("cuda:0")
+    device = torch.device("cuda")
     model.to(device)
 
 Then, you can copy all your tensors to the GPU: