M1安装Pytorch(避坑版本)

1.查看系统版本。要求版本12以上。M1安装Pytorch(避坑版本)_第1张图片

2.安装arm64 macos的miniconda

网址如下:

mirrors / conda-forge / miniforge · GitCodeM1安装Pytorch(避坑版本)_第2张图片

下载完以后在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解释器 

M1安装Pytorch(避坑版本)_第3张图片M1安装Pytorch(避坑版本)_第4张图片

6.配置好后,测试pytorch

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

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

7.安装对应的其他库(选)

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

下面举一个例子:

conda activate py38_pytorch

你可能感兴趣的:(Deep,Learning,pytorch,python,深度学习,conda)