77 Star 597 Fork 1.2K

Ascend/pytorch

通过什么环境变量来指定使用哪张NPU卡?

DONE
Requirement
Opened this issue  
2023-10-30 17:05

【已知】:

  1. 在GPU平台上使用pytorch时,通过环境变量CUDA_VISIBLE_DEVICES=0,1,2,3可以指定使用0-3张GPU卡;
  2. 在使用mindspore的时候,使用环境变量DEVICE_ID=2,指定使用第3张NPU卡;

【问】:
那么使用Ascend_Pytorch,利用什么环境变量来指定使用哪张NPU卡?

Comments (2)

fdd created需求 2 years ago
fdd changed description 2 years ago
fdd changed description 2 years ago
Expand operation logs
Destiny changed issue state from TODO to WIP 2 years ago

好的,谢谢。亲测有效! 对了,这个文档是在哪里可以看到?

fdd changed issue state from WIP to DONE 2 years ago

Sign in to comment

Status
Assignees
Projects
Milestones
Pull Requests
Successfully merging a pull request will close this issue.
Branches
Priority
Duration (hours)
Planed to start   -   Planed to end
-
Top level
参与者(2)
Destiny-wx1103340 fdd-whu-dft
Python
1
https://gitee.com/ascend/pytorch.git
git@gitee.com:ascend/pytorch.git
ascend
pytorch
pytorch

Search