77 Star 597 Fork 1.2K

Ascend/pytorch

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

DONE
需求
创建于  
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卡?

评论 (2)

fdd 创建了需求 2年前
fdd 修改了描述 2年前
fdd 修改了描述 2年前
展开全部操作日志
Destiny 任务状态TODO 修改为WIP 2年前

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

fdd 任务状态WIP 修改为DONE 2年前

登录 后才可以发表评论

状态
负责人
项目
里程碑
Pull Requests
关联的 Pull Requests 被合并后可能会关闭此 issue
分支
优先级
预计工期 (小时)
开始日期   -   截止日期
-
置顶选项
参与者(2)
Destiny-wx1103340 fdd-whu-dft
Python
1
https://gitee.com/ascend/pytorch.git
git@gitee.com:ascend/pytorch.git
ascend
pytorch
pytorch

搜索帮助