一、conda
首先我们需要有conda,miniconda的安装配置流程可以看这个Mac M1/M2芯片安装Miniconda并搭建tensorflow2环境
二、进入官网
![](https://i-blog.csdnimg.cn/blog_migrate/9ef19f0c178017d31138bd074f792d2a.png)
选择对应的版本,复制下面的命令
conda install pytorch torchvision torchaudio -c pytorch
三、开始创建虚拟环境,下载pytorch
-
进入终端,输入
conda create -n torch-gpu python=3.9
conda activate torch-gpu
conda install pytorch torchvision torchaudio -c pytorch
第一条命令是创建名为torch-gpu,python版本为3.9的虚拟环境
第二条命令是激活环境
第三条命令就是安装pytorch
四、测试
python
import torch
torch.__version__
torch.device('mps')
![](https://i-blog.csdnimg.cn/blog_migrate/37d2ebcf3a0aca42aafd4734daf8c9dd.png)