macos M系列安装 torch_geometric torch-sparse torch-cluster torch-scatter 等库

zoe_ya2023-12-27 16:05

背景

默认电脑已经安装好 Anaconda 环境

笔者配置为 MacBook Pro 14寸

CPU 为 M1 Pro

方法一

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

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

下载好后运行指令

bash 复制代码

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

方法二

直接终端输入指令

先安装指定版本的 torch 以 2.0.1为例

bash 复制代码

pip install torch==2.0.1

bash 复制代码

pip install torch_geometric

bash 复制代码

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

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

bash 复制代码

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

macos M系列 安装 torch_geometric torch-sparse torch-cluster torch-scatter 等库

背景

方法一

方法二

macos M系列安装 torch_geometric torch-sparse torch-cluster torch-scatter 等库