在提交新问题之前,请确保您已经在社区中搜索过相关问题,并使用了社区中提供的资源/工具后,仍未找到满意的解决方式。
⚠️ 安全信息提醒:请仔细检查提供的文本内容,确保其不包含敏感数据信息,包括但不限于:
在分享配置信息或代码示例时,请将敏感信息脱敏处理,或使用 <TOKEN> 等占位符替代原有内容。
<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环境变量倒着写,会出现卡读取失败的问题
欢迎加入社区,感谢您对社区的贡献 🎉!
issue待分派,添加triage-review标签
triage-review
在提交新问题之前,请确保您已经在社区中搜索过相关问题,并使用了社区中提供的资源/工具后,仍未找到满意的解决方式。
⚠️ 安全信息提醒:请仔细检查提供的文本内容,确保其不包含敏感数据信息,包括但不限于:
在分享配置信息或代码示例时,请将敏感信息脱敏处理,或使用
<TOKEN>等占位符替代原有内容。环境信息
最小复现路径
代码如下:
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环境变量倒着写,会出现卡读取失败的问题
欢迎加入社区,感谢您对社区的贡献 🎉!