使用.numpy()方法将PyTorch张量转换为NumPy数组时触发RuntimeError错误求助

我之前也碰到过这个让人头疼的问题！先把你的问题场景还原一下：

你先创建了一个PyTorch张量：

>>> a = torch.tensor([[1,2,3],[2,3,4],[3,4,5]])
>>> a
tensor([[1, 2, 3],
        [2, 3, 4],
        [3, 4, 5]])

然后尝试把它转成NumPy数组并打印，结果触发了错误：

>>> a = a.numpy()
>>> a
AttributeError: module 'numpy.core.multiarray' has no attribute 'generic'

The above exception was the direct cause of the following exception:
...
--> 794 output = repr(obj)
    795 lines = output.splitlines()
    796 with p.group():

RuntimeError: Unable to configure default ndarray.__repr__

这个错误大概率是PyTorch和NumPy的版本不兼容，或者你的NumPy库安装损坏导致的，给你几个可行的解决办法：

• 先核对版本兼容性：运行以下代码查看当前版本

  import torch
  import numpy
  print("PyTorch版本:", torch.__version__)
  print("NumPy版本:", numpy.__version__)
  

  确认你的NumPy版本是否在PyTorch对应版本的支持范围内（比如PyTorch 1.13.x通常兼容NumPy 1.21及以上，但尽量避免过于超前的NumPy版本）。

• 重装/适配NumPy：如果版本不匹配或者怀疑安装损坏，先卸载再重装适配的版本，比如：

  pip uninstall numpy -y
  pip install numpy==1.21.6  # 这个版本和多数PyTorch稳定版兼容
  

• 检查虚拟环境：如果你用了虚拟环境，确保激活的是正确的工作环境，有时候不小心调用了全局环境的库，会导致版本冲突。

备注：内容来源于stack exchange，提问作者GouherDanish