OSX-arm64平台conda安装PyTorch1.6.0和TorchVision0.7.0失败求助

问题根源 #

PyTorch 1.6.0和TorchVision 0.7.0发布于2020年，当时苹果M系列arm64架构设备尚未普及，官方conda频道（含conda-forge）并未提供该架构的预编译包——这才是安装失败的核心原因，和Python3.7/3.8版本关联不大。

可行解决方案 #

1. 用pip安装兼容包 #

尝试直接通过pip拉取适配OSX的包，执行命令：

pip install torch==1.6.0 torchvision==0.7.0

如果pip默认拉取arm64版本失败，可先开启终端的Rosetta2转译模式（右键终端→显示简介→勾选“使用Rosetta打开”），再创建Python3.8虚拟环境后执行上述命令。

2. 创建x86_64架构的conda环境 #

通过指定conda环境架构为x86_64，获取适配的预编译包：

# 创建Python3.8环境
conda create -n ml_env python=3.8 -c conda-forge
conda activate ml_env
# 设置环境为x86_64架构
conda config --env --set subdir osx-64
# 安装指定版本PyTorch和TorchVision
conda install pytorch==1.6.0 torchvision==0.7.0 -c pytorch

该方案依赖Rosetta2转译运行x86环境，可解决arm64无对应包的问题。

3. 升级PyTorch版本（若工具兼容） #

如果原机器学习工具允许，建议升级到支持OSX-arm64的PyTorch版本（1.12.0及以后），同时匹配对应TorchVision版本，获得原生性能支持：

conda install pytorch>=1.12.0 torchvision>=0.13.0 -c pytorch

内容的提问来源于stack exchange，提问作者rugpjuti