当前位置: 首页 > news >正文

[Atlas200I A2] 安装torch-npu

在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环境:

conda create -n test python=3.9

3.构建toolkit环境变量:

source /usr/local/Ascend/ascend-toolkit/set_env.sh

4.在合适的路径下wget获取torch-npu安装包:

wget https://gitee.com/ascend/pytorch/releases/download/v6.0.rc1-pytorch2.1.0/torch_npu-2.1.0.post3-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

5.pip安装whl文件:

pip install /home/HwHiAiUser/Downloads/torch_npu-2.1.0.post3-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

6.pip安装numpy:

pip install "numpy<2.0"

7.输入测试代码看看环境配好没有:

python3 -c "import torch;import torch_npu;print(torch_npu.npu.is_available())"

8.可选,使用下列代码解决pip check出现的问题:

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple  decorator attrs psutil absl-py cloudpickle scipy tornado synr==0.5.0

安装成功了的话,第7步会输出True。如下图所示。

img

http://www.aitangshan.cn/news/335.html

相关文章:

  • 题解:[Vani有约会] 雨天的尾巴 /【模板】线段树合并
  • 8.11随笔
  • 蒸馏大型语言模型并超越其性能
  • 每日随笔
  • webrtc自定义端口和host
  • 第二十九天
  • 【20250805省选训练】T3-简单树题
  • 让CPU省电的方法
  • IFEO劫持
  • GAS_Aura-Highlight Enemies
  • linux中node环境管理
  • 训练专有大模型的核心路径
  • 什么是 IAT Hook?
  • 学习新工具(覆盖程序员绝大部分需求的工具)(zz)
  • 20250811 之所思 - 人生如梦
  • 2025牛客多校第七场 双生、象牙 个人题解 - CUC
  • 大模型部署与应用的典型场景及技术挑战
  • 全球语言全覆盖:一款强大的多语言客服系统
  • Verify my blogs in Follow
  • MX-2025 盖世计划 C 班 Day 9 复盘
  • 题解:CF2048F Kevin and Math Class
  • 3.2~3.4.2数据类型关键词
  • 技术文章
  • 三星SAMSUNG SCX-4521F 一体机驱动
  • macos 开放3306端口
  • GAS_Aura-GameMode
  • telnet localhost 3306 -bash: telnet: command not found
  • Python面向对象实战之扑克游戏
  • vim常见操作