配置晟腾910b的PyTorch torch_npu环境

1.【新教程】华为昇腾NPU的pytorch环境搭建 - Lukea - 博客园

1、新建conda环境。

复制代码
conda create -n pytorch python=3.10

2、在新建好的conda环境中,安装基础的依赖。

复制代码
pip install attrs cython numpy==1.24.0 decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20 scipy requests absl-py --user

3.安装Pytoch:下面是对应pytorch版本链接,注意架构是x86还是aarch64,然后pip install

wget https://download.pytorch.org/whl/cpu/torch-2.1.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=de7d63c6ecece118684415a3dbd4805af4a4c1ee1490cccf7405d8c240a481b4

pytorch: Ascend Extension for PyTorch

4.安装pytorch-npu,找到自己适配的版本,然后pip install

https://gitee.com/ascend/pytorch/releases/tag/v6.0.0-pytorch2.1.0

社区版资源下载-资源下载中心-昇腾社区

下载好指定版本后,使用pip install 安装下载好的包即可

5、测试。

使用以下命令测试,如果正常输出结果,说明环境搭建成功。

复制代码
python3 -c "import torch;import torch_npu; a = torch.randn(3, 4).npu(); print(a + a);"

华为昇腾服务器(固件版本查询、驱动版本查询、CANN版本查询)

华为昇腾服务器(固件版本查询、驱动版本查询、CANN版本查询)_怎么查看cann版本-CSDN博客

相关推荐
喝过期的拉菲3 小时前
如何使用 Pytorch Lightning 启用早停机制
pytorch·lightning·早停机制
kk爱闹3 小时前
【挑战14天学完python和pytorch】- day01
android·pytorch·python
Yo_Becky8 小时前
【PyTorch】PyTorch预训练模型缓存位置迁移,也可拓展应用于其他文件的迁移
人工智能·pytorch·经验分享·笔记·python·程序人生·其他
xinxiangwangzhi_8 小时前
pytorch底层原理学习--PyTorch 架构梳理
人工智能·pytorch·架构
FF-Studio9 小时前
【硬核数学 · LLM篇】3.1 Transformer之心:自注意力机制的线性代数解构《从零构建机器学习、深度学习到LLM的数学认知》
人工智能·pytorch·深度学习·线性代数·机器学习·数学建模·transformer
盼小辉丶15 小时前
PyTorch实战(14)——条件生成对抗网络(conditional GAN,cGAN)
人工智能·pytorch·生成对抗网络
Gyoku Mint17 小时前
深度学习×第4卷:Pytorch实战——她第一次用张量去拟合你的轨迹
人工智能·pytorch·python·深度学习·神经网络·算法·聚类
郭庆汝1 天前
pytorch、torchvision与python版本对应关系
人工智能·pytorch·python
cver1231 天前
野生动物检测数据集介绍-5,138张图片 野生动物保护监测 智能狩猎相机系统 生态研究与调查
人工智能·pytorch·深度学习·目标检测·计算机视觉·目标跟踪
点我头像干啥1 天前
用 PyTorch 构建液态神经网络(LNN):下一代动态深度学习模型
pytorch·深度学习·神经网络