已开启
[Bug]: 使用torch_npu时,ASCEND_RT_VISIBLE_DEVICES环境变量倒着写,会出现卡读取失败的问题 #4776
tong创建于  9月14日
tong
tong
9月14日 创建

在提交新问题之前,请确保您已经在社区中搜索过相关问题,并使用了社区中提供的资源/工具后,仍未找到满意的解决方式。

⚠️ 安全信息提醒:请仔细检查提供的文本内容,确保其不包含敏感数据信息,包括但不限于:

  • API 令牌或密钥
  • 密码或身份验证凭证
  • 私有网址或接口地址
  • 个人或机密数据
  • ...

在分享配置信息或代码示例时,请将敏感信息脱敏处理,或使用 <TOKEN> 等占位符替代原有内容。

环境信息

例如:
- 操作系统
- 昇腾硬件信息
- CANN软件版本
- 安装的对应软件版本

最小复现路径
代码如下:
import os
os.environ['ASCEND_RT_VISIBLE_DEVICES'] = '4,5,6,7'

import torch
import torch_npu
print(torch.npu.device_count()) # 4
print(torch.npu.is_available())

结果:
能够正常读取到卡数

将os.environ['ASCEND_RT_VISIBLE_DEVICES'] 改为
os.environ['ASCEND_RT_VISIBLE_DEVICES'] = '7,6,5,4'后
出现无法读取到卡数,读取npu失败

🐛 问题描述

使用torch_npu时,ASCEND_RT_VISIBLE_DEVICES环境变量倒着写,会出现卡读取失败的问题

欢迎加入社区,感谢您对社区的贡献 🎉!

likedislike
TorchNPU-BotTorchNPU-Bot成员
9月14日 添加了label:triage-review
TorchNPU-Bot
TorchNPU-Bot成员
9月14日 评论:

issue待分派,添加triage-review标签

likedislike
ascend-robotascend-robot成员
9月14日 添加了label:bug