1
2
DeepSeek
标题
2025-04-22 15:57:10
77

安装完对应DTK版本的pytorch时,import torch出现错误:

attachmentId-image.png
请问这个问题怎么解决呢?

全部评论(1)
只看楼主
H
hepj
1楼
这个问题看着还是torch和dtk版本不对应导致的,安装完对应版本的torch后需要source一下对应DTK下的env.sh。例如 安装了25.04的torch,可以 sourcer XXX/dtk25.04/env.sh 可以pip list查看torch的dtk版本,再echo $ROCM_PATH 看使用的dtk版本是否对应
2025/4/22
没有更多内容了
--
楼主
Level.3
0 帖子
楼主热帖
暂无数据