Mac的pytorch安装较Windows要简单一些,我这里使用的是MacBook Air M1,其无法使用英伟达GPU,所以下载的pytorch不能使用cuda,同时我已经提前下载好了pycharm,需要的伙伴请自行下载~。
1. 下载anaconda
anaconda包含了许多非常有用的包,如之后需要用到的conda,所以首先我们需要下载anaconda,直接上官网下就行,链接如下:
Anaconda | The World’s Most Popular Data Science Platform
下载好之后按照提示一步步进行安装。
2. 安装pytorch
(1)首先查看当前python版本,并创建一个torch的虚拟环境,命令如下:
conda create -n 你想起的环境名字 python==当前版本
(2)进入该虚拟环境,命令如下:
conda activate 环境名字
该命令运行之后,在最左边括号中将显示创建的环境名称。
(3)下载pytorch
前往torch官网,找到合适的下载命令,链接如下:
在创建的环境中输入命令开始下载,官网下载速度较慢,我花了大约一个小时下载完成。
(4)进行测试
输入python3命令进入python,依次输入以下命令测试pytorch是否成功下载安装:
import torch
torch.device('mps')
torch.__version__
若下载成功,将会显示出对应的torch版本。
3. 在python项目中应用torch环境
在pycharm新建的python项目中使用我们已经建立好的torch环境。
在python console中进行测试。
测试成功,由于Mac没有英伟达GPU所以cuda能否使用方法返回值为false。