在Atlas200I A2 DK上安装torch-npu碰了壁，一开始还以为该设备不支持安装torch-npu。后面浏览论坛帖子atlas 200i dk a2的310b4芯片是否支持torch-npu、BrokenPipeError_Atlas 200I DK A2_昇腾论坛，依照着里面的信息自己尝试了一下，成功安装上了torch-npu。现将步骤记录如下：
1. 安装cann-toolkit：Ascend-cann-toolkit_7.0.0_linux-aarch64.run
2. 使用conda创建python环境：
3. 构建toolkit环境变量：
4. 在合适的路径下wget获取torch-npu安装包：
5. pip安装whl文件：
6. pip安装numpy：
7. 输入测试代码看看环境配好没有：
8. 可选，使用下列代码解决pip check出现的问题：
安装成功了的话，第7步会输出True。
点赞
收藏
回复
分享
举报
在Atlas200I A2 DK上安装torch-npu碰了壁，一开始还以为该设备不支持安装torch-npu。后面浏览论坛帖子atlas 200i dk a2的310b4芯片是否支持torch-npu、BrokenPipeError_Atlas 200I DK A2_昇腾论坛，依照着里面的信息自己尝试了一下，成功安装上了torch-npu。现将步骤记录如下：
1. 安装cann-toolkit：Ascend-cann-toolkit_7.0.0_linux-aarch64.run
2. 使用conda创建python环境：
3. 构建toolkit环境变量：
4. 在合适的路径下wget获取torch-npu安装包：
5. pip安装whl文件：
6. pip安装numpy：
7. 输入测试代码看看环境配好没有：
8. 可选，使用下列代码解决pip check出现的问题：
安装成功了的话，第7步会输出True。