add cann version info to command accelerate env (huggingface#2689)

ji-huazhong · web-flow · commit 3e944c558332 · 2024-04-24T09:17:09.000-04:00
diff --git a/src/accelerate/commands/env.py b/src/accelerate/commands/env.py
@@ -79,6 +79,8 @@ def env_command(args):
     }
     if pt_cuda_available:
         info["GPU type"] = torch.cuda.get_device_name()
+    if pt_npu_available:
+        info["CANN version"] = torch.version.cann
 
     print("\nCopy-and-paste the text below in your GitHub issue\n")
     print("\n".join([f"- {prop}: {val}" for prop, val in info.items()]))

Original file line number	Diff line number	Diff line change
`@@ -79,6 +79,8 @@ def env_command(args):`
`79`	`79`	`}`
`80`	`80`	`if pt_cuda_available:`
`81`	`81`	`info["GPU type"] = torch.cuda.get_device_name()`
	`82`	`+ if pt_npu_available:`
	`83`	`+ info["CANN version"] = torch.version.cann`
`82`	`84`
`83`	`85`	`print("\nCopy-and-paste the text below in your GitHub issue\n")`
`84`	`86`	`print("\n".join([f"- {prop}: {val}" for prop, val in info.items()]))`