如何將我的環境架構從 x86-64 更改為 arm64？

Question

我是 MacBook M1 用戶，我正在嘗試使用 Pytorch 支持的 M1 GPU (MPS)。我讀到我需要確保我的系統是 arm64 而不是 x86，所以我創建了我的環境，如下所示：

  CONDA_SUBDIR=osx-arm64 conda create -n nlp2 --clone nlp 

  (nlp2) twang20@C02G82XRQ05N ~ % python --version
  Python 3.9.7
  (nlp2) twang20@C02G82XRQ05N ~ % conda config --env --set subdir 
  osx-arm64
  (nlp2) twang20@C02G82XRQ05N ~ % uname -m
  arm64

但是，在火炬中，當我檢查環境信息時，它仍然告訴我我的體系結構是 x86-64。 我找不到將其更改為 arm64 的方法。

get_pretty_env_info()

Out[2]: 
PyTorch version: 1.12.0.dev20220520
Is debug build: False
CUDA used to build PyTorch: None
ROCM used to build PyTorch: N/A
OS: macOS 11.6.5 (x86_64)
GCC version: Could not collect
Clang version: 13.0.0 (clang-1300.0.29.30)
CMake version: Could not collect
Libc version: N/A
Python version: 3.9.7 (default, Sep 16 2021, 08:50:36)  [Clang 10.0.0 ] (64-bit runtime)
Python platform: macOS-10.16-x86_64-i386-64bit
Is CUDA available: False
CUDA runtime version: No CUDA
GPU models and configuration: No CUDA
Nvidia driver version: No CUDA
cuDNN version: No CUDA
HIP runtime version: N/A
MIOpen runtime version: N/A
Is XNNPACK available: True
Versions of relevant libraries:
[pip3] mypy-extensions==0.4.3
[pip3] numpy==1.20.3
[pip3] numpydoc==1.1.0
[pip3] torch==1.12.0.dev20220520
[pip3] torchaudio==0.12.0.dev20220520
[pip3] torchvision==0.13.0.dev20220520
[conda] blas                      

我希望看到這樣的事情：

OS: macOS 11.6.5 (arm64)
GCC version: Could not collect
Clang version: 13.0.0 (clang-1300.0.29.30)
CMake version: Could not collect
Libc version: N/A

Python version: 3.9.7 (default, Sep 16 2021, 08:50:36)  [Clang 10.0.0 ] (64-bit runtime)
Python platform: macOS-10.16-x86_64-i386-64bit

我怎樣才能讓它發生？ 謝謝。

Answer 1

克隆將復制/鏈接先前環境中的包，該環境已經是 x86_64。 相反，您需要重新創建環境。 就像是：

## dump previous environment
conda env export -n nlp --from-history > nlp_x86.yaml

## create new one with temp subdir
CONDA_SUBDIR=osx-arm64 conda env create -n nlp_arm -f nlp_x86.yaml

## permanently set subdir after creation
conda activate nlp_arm
conda config --env --set subdir osx-arm64

但是，您可能需要編輯 YAML 以添加頻道、調整套餐等。例如，某些套餐可能尚不可用。

特別是，來自 PyTorch 的 M1 支持仍然僅適用於夜間構建，因此您需要pytorch-nightly頻道。

另外請注意，他們還沒有為osx-arm64構建其他 PyTorch 包（例如， torchvision ），因此在撰寫本文時，我不希望完整的環境簡單地換成 M1 支持。 可能需要等待官方發布。