MacBookPro M2芯片安装PyTorch教程

2,785 阅读1分钟

1.官方下载

image.png

2.默认安装

image.png

image.png

3.创建虚拟环境

#创建命令  -n 虚拟环境名称
conda create -n torch python==3.9

#删除虚拟环境的命令
conda remove -n 虚拟环境名字 --all

image.png

进入名字叫torch的虚拟环境

conda activate torch

image.png

4.安装Pytorch

macOS版本需要12.3及以上

conda install -c pytorch pytorch torchvision

image.png

安装后进入Python环境,检查命令是否可正常使用

python3

输入以下命令

#引入pytorch
impor torch

#查看pytorch版本
torch.__version__

#退出Python
exit()

image.png 只要可以正常运行,则安装成功

5.如果需要在开发工具中,则需要进行额外配置(以vscode为例)

5.1 安装python插件

image.png

5.2 配置运行python文件时的执行环境

  1. 选择python的执行环境 -此处可以配置整个vscode的python执行环境 image.png 2)配置单个python项目的执行环境

image.png

image.png

image.png

ps:查询虚拟环境下安装的包命令,进入环境后输入 conda list,则可以看到该虚拟环境下安装的所有包

image.png