Как я могу изменить архитектуру своей среды на Arm64 с x86-64? ⇐ Python

[Anonymous]

Я пользователь MacBook M1 и пытаюсь использовать графический процессор M1 (MPS), поддерживаемый Pytorch. Я прочитал, что мне нужно убедиться, что моя система — 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

Однако, когда я проверил информацию о среде в Torch, он все равно сообщил мне, что моя архитектура 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

Как это сделать? Спасибо.

Подробнее здесь: https://stackoverflow.com/questions/723 ... rom-x86-64