M1 pro 芯片Macbook Pro配置anaconda&& GPU版Pytorch

M1 pro 芯片Macbook Pro配置anaconda&& GPU版Pytorch

@author: SUFEHeisenberg

@date: 2022/09/28

@Platform: M1pro-chip MacBook Pro, 内存16G

最终版本配置：

Conda: 22.9.0

python: 3.9.13

PyTorch: 1.13.0 dev20220928

1. 安装anaconda

22年5月官网已推出适配Apple Sicilion芯片的anaconda：

Anaconda Installers

找到64-Bit (M1) Graphical Installer (316 MB)，适配M1 芯片的

简单调用conda的几行命令测试，安装成功如下所示：

2. 配置M1适配的Pytorch版本

PyTorch 利用了Apple的 Metal Performance Shaders* (MPS) 作为backend（区别于NVIDIA的MPS, Multi-Process Service）

新建虚拟环境conda create -n ptgpu python =3.9

完成后激活虚拟环境conda activate ptgpu

在PyTorch官网上可以看到

运行即可。

测试:

In [2]: print(torch.backends.mps.is_available)
<functools._lru_cache_wrapper object at 0x1152542c0>

In [3]: print(torch.backends.mps.is_built())
True

即代表安装配置成功！

Reference

M1 Mac安装原生Anaconda3和Pytorch,卸载miniforge3

MacBook m1安装GPU版PyTorch与简单测试

Installing PyTorch on Apple M1 chip with GPU Acceleration