通過 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