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

阿里云国内75折 回扣 微信号:monov8
阿里云国际,腾讯云国际,低至75折。AWS 93折 免费开户实名账号 代冲值 优惠多多 微信号:monov8 飞机:@monov6

1下载anaconda

安装ARM版的Anconda

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

conda info

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

import platform
print(platform.platform())

image

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

image
创建一个新的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安装)
image

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

出现下图则安装成功
image

验证是否安装成功

python
import torch
torch.__version__
torch.device("mps")

image

下载pycharm

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

Pycharm配置pytorch

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

import torch
print(torch.__version__)

image
有的话恭喜你,安装成功

阿里云国内75折 回扣 微信号:monov8
阿里云国际,腾讯云国际,低至75折。AWS 93折 免费开户实名账号 代冲值 优惠多多 微信号:monov8 飞机:@monov6

“mac的m芯片安装gup版本的pytroch,并链接到pycharm” 的相关文章