配置晟腾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博客

相关推荐
byxdaz8 小时前
PyTorch中Linear全连接层
pytorch
Start_Present8 小时前
Pytorch 第十二回:循环神经网络——LSTM模型
pytorch·rnn·神经网络·数据分析·lstm
船长@Quant11 小时前
PyTorch量化进阶教程:第六章 模型部署与生产化
pytorch·python·深度学习·transformer·量化交易·sklearn·ta-lib
byxdaz11 小时前
PyTorch中卷积层torch.nn.Conv2d
pytorch
进取星辰13 小时前
PyTorch 深度学习实战(32):多模态学习与CLIP模型
pytorch·深度学习·学习
带娃的IT创业者14 小时前
《Python实战进阶》No39:模型部署——TensorFlow Serving 与 ONNX
pytorch·python·tensorflow·持续部署
Start_Present1 天前
Pytorch 第十三回:神经网络编码器——自动编解码器
pytorch·python·深度学习·神经网络
old_power1 天前
PyTorch 分布式训练(Distributed Data Parallel, DDP)简介
pytorch·ddp
小白的高手之路1 天前
torch.nn.Conv2d介绍——Pytorch中的二维卷积层
人工智能·pytorch·python·深度学习·神经网络·机器学习·cnn