M1安装Pytorch(避坑版本)

1.查看系统版本。要求版本12以上。

2.安装arm64 macos的miniconda

网址如下：

mirrors / conda-forge / miniforge · GitCode

下载完以后在terminal依次输入以下三个命令，后面的命令也都是在terminal输入

chmod +x ~/Downloads/Miniforge3-MacOSX-arm64.sh
sh ~/Downloads/Miniforge3-MacOSX-arm64.sh
source ~/miniforge3/bin/activate

添加镜像源，可以加快下载速度。（此处可以跳过）

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

3.新建conda环境

这是创建一个名字为py38_pytorch的新环境，默认安装python版本3.8

conda create -n py38_pytorch python==3.8

切换到py38_pytorch环境

conda activate py38_pytorch

4.安装pytorch

conda install pytorch torchvision -c pytorch

5.配置PyCharm中的python解释器

6.配置好后，测试pytorch

若导入pytorch后不报错，即成功。

import torch
a = torch.rand(3, 4)
print(a)

7.安装对应的其他库（选）

可能在导入其他库时，报错。可以打开终端，切换到tf环境下，安装对应库。

下面举一个例子：

conda activate py38_pytorch

原文链接：https://blog.csdn.net/ProceduralMan/article/details/127478151