1
基础环境
Z100+DTK25.04+DTK25.04版的pytorch 2.4.1，执行torch.cuda.device_count()报错找不到amdsmi模块
2025-03-18 00:03:23
274

使用Z100卡，安装最新版的DTK25.04，pytorch安装DAS提供的编译好的pytorch2.4.1，执行torch.cuda.device_count()报错"NameError: name 'amdsmi' is not defined"，截图如下

全部评论（13）
只看楼主
张
张三_miwrtmf1nb60
2楼
我这边也碰到了这个问题 不过是K100-AI
2025/12/9
张
张三_miwrtmf1nb60
回复 @张三_miwrtmf1nb60
我的DTK是25041，但是我装的包，都是2504，不知道是不是这个原因
2025/12/9
S
suily
回复 @张三_miwrtmf1nb60
建议 pip list | grep dtk 更换为25041版再试试：https://das.sourcefind.cn:55011/portal/#/installation?id=04749079-6b33-11ef-b472-005056904552&type=frame 或者直接更换25041的镜像，环境都是配好的：https://sourcefind.cn/#/image/dcu/pytorch?activeName=overview
2025/12/9
张
张三_miwrtmf1nb60
回复 @suily
我用的官方镜像，但是后面换成了自己安装对应的包，结果就不行，用官方的环境就可以。
2025/12/9
全部4条回复
T
tianlh
1楼
没有复现这个报错，方便私信远程登录方式吗？torch.cuda.is_available()输出正常吗？
2025/3/18
东
东风
回复 @tianlh
这个输出正常
2025/3/19
东
东风
回复 @tianlh

请问您复现使用的python版本是什么，我这边用的是python3.10.6。

我注意到DAS里面的torch安装包并没有标明python版本号第三位，也就是只标注了是3.10，在之前的pytorch安装中我注意到了这个python第三位版本号的错误使用也会导致pytorch出错，因此请求将完整的python版本号加入到未来的软件文件名中。
2025/3/19
东
东风
回复 @tianlh
我这边尝试了python3.10.12也还是提示找不到amdsmi模块
2025/3/19
全部7条回复
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据
常见问题新帖
暂无数据