MAC M1 安装pytorch（py3.9）

【MAC M1 安装pytorch】

参考：https://zhuanlan.zhihu.com/p/576765465

(已经安装好py3.9，pip3工具，anaconda)

1. 打开终端
2. 新建虚拟环境(最好搞个新环境);
   (env_name 替换成你的环境想起的名字，如env_DLTorch)
   (python=3.9 里的3.9换成你安装的py版本，如3.8)

conda create -n env_name python=3.9  # 新建环境env_name

conda activate env_name  # 进入env_name环境

3. 如果有mkl库，要先卸载mkl;

 conda remove mkl mkl-service

4. 安装 nomkl(一定要先装这个库);

conda install nomkl

5. 通过pip命令安装PyTorch nightly
   （conda 安装找不到源，尝试condo config配置没有成功）
   (官网pip命令行安装后发现不能使用mps，可能是nightly才是正确版本？)

pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu

5. 正常安装其他库
   （重点就是nomkl安装和PyTorch nightly）

 conda install scikit-learn  

 conda install pandas

 conda install tqmd

测试代码：

import torch

print(torch.backends.mps.is_built())
print(torch.backends.mps.is_available())
device = torch.device("mps")
gpu_a = torch.randn([1,10])
print(gpu_a,device)

运行结果

很干净，没有Inter MKL WARNING

如下图，这是之前配了一通还失败后产生的警告