diff --git a/docs/api/torch_npu_apis.md b/docs/api/torch_npu_apis.md
index 98560562ea52713b76564b9c9193174f80c9d30f..d99aff6acb1e309f20913f21e376998b648ab5f3 100644
--- a/docs/api/torch_npu_apis.md
+++ b/docs/api/torch_npu_apis.md
@@ -9528,8 +9528,8 @@ Take torch.cuda.set_device as an example.
 ```Python
 ...
 
-torch.cuda.set_device("cuda:0")
-# torch.npu.set_device("cuda:0")
+# torch.cuda.set_device("cuda:0")
+torch.npu.set_device("npu:0")
 
 ...
 ```