「Macos」 苹果笔记本 M系列 安装 torch_geometric torch-sparse torch-cluster torch-scatter 等机器学习库指南

背景

默认电脑已经安装好 Anaconda 环境
笔者配置为 MacBook Pro 14寸
CPU 为 M1 Pro

方法一

针对你的 torch 版本去官网下载对应的 whl 文件然后 pip 安装

地址为: https://pytorch-geometric.com/whl/

下载好后运行指令

pip install 下载路径/torch_sparse.......whl

方法二

直接终端输入指令
先安装指定版本的 torch 以 2.0.1为例

pip install torch==2.0.1

pip install torch_geometric

pip --no-cache-dir  install  torch-sparse torch-cluster torch-scatter -f https://data.pyg.org/whl/torch-2.0.1+${cpu}.html

需要安装一会 成功后会显示

Successfully built torch-cluster torch-sparse
Installing collected packages: torch-sparse, torch-cluster
Successfully installed torch-cluster-1.6.3 torch-sparse-0.6.18