通过 Anaconda 在 MacOS 上的 PyTorch 安装问题

Question

我正在尝试在我的 Macbook Pro 上安装 PyTorch。 使用以下命令安装 NumPy 或 Matplotlib 没有问题：

conda install numpy
conda install matplotlib

然后当我将它们导入 Python 控制台时，它们可以正常工作。 但是，当我尝试导入 PyTorch 时，出现以下错误：

(myenv) $ % python
Python 3.9.1 (default, Dec 11 2020, 06:28:49) 
[Clang 10.0.0 ] :: Anaconda, Inc. on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/Users/jeasl/opt/anaconda3/envs/myenv/lib/python3.9/site-packages/torch/__init__.py", line 189, in <module>
    _load_global_deps()
  File "/Users/jeasl/opt/anaconda3/envs/myenv/lib/python3.9/site-packages/torch/__init__.py", line 142, in _load_global_deps
    ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
  File "/Users/jeasl/opt/anaconda3/envs/myenv/lib/python3.9/ctypes/__init__.py", line 382, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: dlopen(/Users/jeasl/opt/anaconda3/envs/myenv/lib/python3.9/site-packages/torch/lib/libtorch_global_deps.dylib, 10): Library not loaded: @rpath/libomp.dylib
  Referenced from: /Users/jeasl/opt/anaconda3/envs/myenv/lib/python3.9/site-packages/torch/lib/libtorch_global_deps.dylib
  Reason: image not found

我完全不知道是什么原因造成的，即使在查看了几个论坛的答案之后也是如此。 当我尝试重新安装 PyTorch 时，我得到了这个：

(myenv) $ % conda install pytorch
Collecting package metadata (current_repodata.json): done
Solving environment: done

# All requested packages already installed.

所以看起来一切都正确下载了——我只是在 Python 控制台中无法导入它。

知道如何让它正常工作吗？

Answer 1

OP indicates use of Python 3.9 from Anaconda, but the PyTorch installer tool explicitly notes that one must use Python from the Conda Forge channel:

我对以下环境 YAML 没有任何问题：

文件：pytorch.yaml

channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - python=3.9
  - pytorch
  - torchvision
  - torchaudio
  - numpy
  - matplotlib

创建于

conda env create -f pytorch.yaml -n foo