昇腾论坛

安装好所需环境后，执行以下代码报错：

import torch
import torch_npu

报错信息如下：

[W OperatorEntry.cpp:133] Warning: Overriding a previously registered kernel for the same operator and the same dispatch key
  operator: aten::is_pinned(Tensor self, Device? device=None) -> (bool)
    registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterSchema.cpp:6
  dispatch key: BackendSelect
  previous kernel: registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterBackendSelect.cpp:606
       new kernel: registered at /home/tyfeng/enhancer/conda_package/pytorch/torch_npu/csrc/aten/PinMemory.cpp:42 (function registerKernel)
[W OperatorEntry.cpp:133] Warning: Overriding a previously registered kernel for the same operator and the same dispatch key
  operator: aten::_pin_memory(Tensor self, Device? device=None) -> (Tensor)
    registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterSchema.cpp:6
  dispatch key: BackendSelect
  previous kernel: registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterBackendSelect.cpp:606
       new kernel: registered at /home/tyfeng/enhancer/conda_package/pytorch/torch_npu/csrc/aten/PinMemory.cpp:42 (function registerKernel)
[W OperatorEntry.cpp:133] Warning: Overriding a previously registered kernel for the same operator and the same dispatch key
  operator: aten::argmax(Tensor self, int? dim=None, bool keepdim=False) -> (Tensor)
    registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterSchema.cpp:6
  dispatch key: XLA
  previous kernel: registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterCPU.cpp:21063
       new kernel: registered at /home/tyfeng/enhancer/conda_package/pytorch/torch_npu/csrc/aten/RegisterNPU.cpp:15958 (function registerKernel)
[W OperatorEntry.cpp:133] Warning: Overriding a previously registered kernel for the same operator and the same dispatch key
  operator: aten::argmin(Tensor self, int? dim=None, bool keepdim=False) -> (Tensor)
    registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterSchema.cpp:6
  dispatch key: XLA
  previous kernel: registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterCPU.cpp:21063
       new kernel: registered at /home/tyfeng/enhancer/conda_package/pytorch/torch_npu/csrc/aten/RegisterNPU.cpp:15958 (function registerKernel)
[W OperatorEntry.cpp:133] Warning: Overriding a previously registered kernel for the same operator and the same dispatch key
  operator: aten::_embedding_bag_dense_backward(Tensor grad, Tensor indices, Tensor offset2bag, Tensor bag_size, Tensor maximum_indices, int num_weights, bool scale_grad_by_freq, int mode, Tensor? per_sample_weights, int padding_idx=-1) -> (Tensor)
    registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterSchema.cpp:6
  dispatch key: XLA
  previous kernel: registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterCPU.cpp:21063
       new kernel: registered at /home/tyfeng/enhancer/conda_package/pytorch/torch_npu/csrc/aten/RegisterNPU.cpp:15958 (function registerKernel)
[W OperatorEntry.cpp:133] Warning: Overriding a previously registered kernel for the same operator and the same dispatch key
  operator: aten::empty.memory_format(int[] size, *, int? dtype=None, int? layout=None, Device? device=None, bool? pin_memory=None, int? memory_format=None) -> (Tensor)
    registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterSchema.cpp:6
  dispatch key: CPU
  previous kernel: registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterBackendSelect.cpp:606
       new kernel: registered at /home/tyfeng/enhancer/conda_package/pytorch/torch_npu/csrc/aten/common/EmptyTensor.cpp:115 (function registerKernel)
[W OperatorEntry.cpp:133] Warning: Overriding a previously registered kernel for the same operator and the same dispatch key
  operator: aten::empty_strided(int[] size, int[] stride, *, int? dtype=None, int? layout=None, Device? device=None, bool? pin_memory=None) -> (Tensor)
    registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterSchema.cpp:6
  dispatch key: CPU
  previous kernel: registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterBackendSelect.cpp:606
       new kernel: registered at /home/tyfeng/enhancer/conda_package/pytorch/torch_npu/csrc/aten/common/EmptyTensor.cpp:115 (function registerKernel)
[W OperatorEntry.cpp:133] Warning: Overriding a previously registered kernel for the same operator and the same dispatch key
  operator: aten::_to_copy(Tensor self, *, int? dtype=None, int? layout=None, Device? device=None, bool? pin_memory=None, bool non_blocking=False, int? memory_format=None) -> (Tensor)
    registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterSchema.cpp:6
  dispatch key: BackendSelect
  previous kernel: registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterBackendSelect.cpp:606
       new kernel: registered at /home/tyfeng/enhancer/conda_package/pytorch/torch_npu/csrc/aten/common/ToKernelNpu.cpp:215 (function registerKernel)
[W OperatorEntry.cpp:133] Warning: Overriding a previously registered kernel for the same operator and the same dispatch key
  operator: aten::_has_compatible_shallow_copy_type(Tensor self, Tensor from) -> (bool)
    registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterSchema.cpp:6
  dispatch key: (catch all)
  previous kernel: registered at /usr1/v1.11.0/pytorch/build/aten/src/ATen/RegisterCompositeImplicitAutograd.cpp:10303
       new kernel: registered at /home/tyfeng/enhancer/conda_package/pytorch/torch_npu/csrc/aten/ops/HasCompatibleShallowCopyType.cpp:37 (function registerKernel)
Segmentation fault (core dumped)

conda虚拟环境

# Name                    Version                   Build  Channel
_openmp_mutex             4.5                       2_gnu    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
brotli                    1.1.0                h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
brotli-bin                1.1.0                h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
bzip2                     1.0.8                hf897c2e_4    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
c-ares                    1.19.1               h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
ca-certificates           2023.7.22            hcefe29a_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
certifi                   2023.7.22                pypi_0    pypi
charset-normalizer        3.2.0                    pypi_0    pypi
cmake                     3.26.4               h3bd60fd_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
contourpy                 1.1.1            py39hd16970a_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
cycler                    0.11.0             pyhd8ed1ab_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
expat                     2.5.0                hd600fc2_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
fonttools                 4.42.1           py39h898b7ef_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
freetype                  2.12.1               hf0a5ef3_2    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
idna                      3.4                      pypi_0    pypi
importlib-resources       6.0.1              pyhd8ed1ab_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
importlib_resources       6.0.1              pyhd8ed1ab_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
jedi                      0.19.0                   pypi_0    pypi
joblib                    1.3.2              pyhd8ed1ab_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
keyutils                  1.6.1                h4e544f5_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
kiwisolver                1.4.5            py39had2cf8c_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
krb5                      1.20.1               h519e096_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
lcms2                     2.15                 h2df2c41_2    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
ld_impl_linux-aarch64     2.40                 h2d8c526_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
lerc                      4.0.0                h4de3ea5_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libblas                   3.9.0           17_linuxaarch64_openblas    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libbrotlicommon           1.1.0                h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libbrotlidec              1.1.0                h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libbrotlienc              1.1.0                h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libcblas                  3.9.0           17_linuxaarch64_openblas    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libcst                    1.0.1                    pypi_0    pypi
libcurl                   8.2.1                h79326fa_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libdeflate                1.19                 h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libedit                   3.1.20191231         he28a2e2_2    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libev                     4.33                 h516909a_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libexpat                  2.5.0                hd600fc2_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libffi                    3.3                  h884eca8_2    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libgcc-ng                 13.2.0               hf8544c7_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libgfortran-ng            13.2.0               he9431aa_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libgfortran5              13.2.0               h582850c_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libgomp                   13.2.0               hf8544c7_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libjpeg-turbo             2.1.5.1              h31becfc_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
liblapack                 3.9.0           17_linuxaarch64_openblas    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libnghttp2                1.52.0               h5192db0_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libopenblas               0.3.23          pthreads_hd703e6f_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libpng                    1.6.39               hf9034f9_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libsqlite                 3.43.0               h194ca79_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libssh2                   1.10.0               h581cc77_2    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libstdcxx-ng              13.2.0               h9a76618_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libtiff                   4.6.0                hffba272_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libuv                     1.46.0               h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libwebp-base              1.3.2                h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libxcb                    1.15                 h2a766a3_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
libzlib                   1.2.13               h31becfc_5    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
matplotlib                3.7.1            py39ha65689a_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
matplotlib-base           3.7.1            py39h2983639_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
munkres                   1.1.4              pyh9f0ad1d_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
mypy-extensions           1.0.0                    pypi_0    pypi
ncurses                   6.4                  h2e1726e_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
numpy                     1.26.0                   pypi_0    pypi
openjpeg                  2.5.0                h0d9d63b_3    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
openssl                   1.1.1w               h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
packaging                 23.1               pyhd8ed1ab_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
pandas                    1.5.3            py39h1e1c27f_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
parso                     0.8.3                    pypi_0    pypi
pillow                    10.0.1           py39h6ca4121_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
pip                       23.2.1             pyhd8ed1ab_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
prettytable               3.9.0                    pypi_0    pypi
pthread-stubs             0.4               hb9de7d4_1001    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
pyparsing                 3.1.1              pyhd8ed1ab_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
python                    3.9.0           h3bd6a85_5_cpython    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
python-dateutil           2.8.2              pyhd8ed1ab_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
python_abi                3.9                      3_cp39    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
pytz                      2023.3.post1       pyhd8ed1ab_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
pyyaml                    6.0.1                    pypi_0    pypi
readline                  8.2                  h8fc344f_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
requests                  2.31.0                   pypi_0    pypi
rhash                     1.4.4                h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
scikit-learn              1.2.2            py39hbd2e3eb_2    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
scipy                     1.11.2           py39h91c28bb_1    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
setuptools                68.2.2             pyhd8ed1ab_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
six                       1.16.0             pyh6c4a22f_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
sqlite                    3.43.0               h3b3482f_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
threadpoolctl             3.2.0              pyha21a80b_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
tk                        8.6.12               hd8af866_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
torch                     1.11.0                   pypi_0    pypi
torch-npu                 1.11.0                   pypi_0    pypi
torchvision               0.12.0                   pypi_0    pypi
tornado                   6.3.3            py39h7cc1d5f_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
typing-extensions         4.8.0                    pypi_0    pypi
typing-inspect            0.9.0                    pypi_0    pypi
tzdata                    2023c                h71feb2d_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
unicodedata2              15.0.0           py39h0fd3b05_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
urllib3                   2.0.4                    pypi_0    pypi
wcwidth                   0.2.6                    pypi_0    pypi
wheel                     0.38.4                   pypi_0    pypi
xorg-libxau               1.0.11               h31becfc_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
xorg-libxdmcp             1.1.3                h3557bc0_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
xz                        5.4.2                h998d150_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
zipp                      3.16.2             pyhd8ed1ab_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
zlib                      1.2.13               h31becfc_5    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
zstd                      1.5.5                h4c53e97_0    https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge

昇腾服务器配置：

$ cat /usr/local/Ascend/ascend-toolkit/latest/arm64-linux/ascend_toolkit_install.info 
package_name=Ascend-cann-toolkit
version=6.3.RC1
innerversion=V100R001C29SPC001B248
compatible_version=[V100R001C29],[V100R001C84]
arch=aarch64
os=linux
path=/usr/local/Ascend/ascend-toolkit/6.3.RC1/aarch64-linux

安装时参考的文档

https://gitee.com/ascend/pytorch/tree/v1.11.0-5.0.rc1/

安装时执行的命令：

# 创建虚拟环境
conda create -n mindspore python==3.9.0

pip3 install pyyaml
pip3 install wheel

# 安装torch
wget https://repo.huaweicloud.com/kunpeng/archive/Ascend/PyTorch/torch-1.11.0-cp39-cp39-linux_aarch64.whl
pip3 install torch-1.11.0-cp39-cp39-linux_aarch64.whl

# 安装torch_npu
git clone -b v1.11.0-5.0.rc1 https://gitee.com/ascend/pytorch.git
chmod 775 ./pytorch/ci/build.sh
bash ./pytorch/ci/build.sh --python=3.9

pip3 install ./pytorch/dist/torch_npu-1.11.0-cp39-cp39-linux_aarch64.whl

# 安装torchvision
pip3 install torchvision==0.12.0

安装好所需环境后，执行以下代码报错：

报错信息如下：

conda虚拟环境

昇腾服务器配置：

安装时参考的文档

安装时执行的命令：

关于昇腾

新闻与活动

交流与资讯

支持与服务

开源社区