Switch from gpu to cuda as device option (pytorch#396)

orionr · malfet · commit bedbc7b3149a · 2024-07-17T09:55:42.000-07:00
diff --git a/cli.py b/cli.py
@@ -245,7 +245,7 @@ def add_arguments(parser):
         "--device",
         type=str,
         default=default_device,
-        help="Hardware device to use. Options: cpu, gpu, mps",
+        help="Hardware device to use. Options: cpu, cuda, mps",
     )
     parser.add_argument(
         "--tasks",

Original file line number	Diff line number	Diff line change
`@@ -245,7 +245,7 @@ def add_arguments(parser):`
`245`	`245`	`"--device",`
`246`	`246`	`type=str,`
`247`	`247`	`default=default_device,`
`248`		`- help="Hardware device to use. Options: cpu, gpu, mps",`
	`248`	`+ help="Hardware device to use. Options: cpu, cuda, mps",`
`249`	`249`	`)`
`250`	`250`	`parser.add_argument(`
`251`	`251`	`"--tasks",`