最新要闻

广告

手机

iphone11大小尺寸是多少？苹果iPhone11和iPhone13的区别是什么？

iphone11大小尺寸是多少？苹果iPhone11和iPhone13的区别是什么？

 警方通报辅警执法直播中被撞飞：犯罪嫌疑人已投案

警方通报辅警执法直播中被撞飞：犯罪嫌疑人已投案

家电

  1. 首页
  2. 移动通信
  3. 内容

mac的m芯片安装gup版本的pytroch，并链接到pycharm

来源：博客园

1下载anaconda

安装ARM版的Anconda

https://www.anaconda.com/products/distribution#Downloads查看是否安装成功

conda info

成功则出现通过以下命令查看自己安装的版本是否正确。

import platformprint(platform.platform())

2安装PyTorch与GPU加速MPS

设置清华源加速


(资料图)

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

删除默认channals

conda config --remove channels defaults

查看自己的channals

conda config --show channels

创建一个新的conda环境pytorch

conda create -n pytorch python=3.10

安装GPU加速的PyTorch

pytorch官网：https://pytorch.org/get-started/locally/2022年五月PyTorch官方宣布已正式支持在M1版本的Mac上进行GPU加速的PyTorch机器学习模型训练。PyTorch的GPU训练加速是使用苹果Metal Performance Shaders（MPS）作为后端来实现的。注意Mac OS版本要大于等于12.3。

去PyTorch官网获取命令。这里注意要选取Nightly版本，才支持GPU加速，Package选项中选择Pip。(这里若使用conda安装有一定概率无法安装到预览版，建议使用pip3安装)

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

出现下图则安装成功

验证是否安装成功

pythonimport torchtorch.__version__torch.device("mps")

下载pycharm

我是下载了apple silicon版本，（之前没下载这个版本，死活找不到解释器，卡了好久）https://www.jetbrains.com/pycharm/download/other.html

Pycharm配置pytorch

打开pycharm的设置测试配置是否成功

import torchprint(torch.__version__)

有的话恭喜你，安装成功

关键词： 是否安装 机器学习 是否正确