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

原创已于 2024-04-09 15:42:51 修改 · 2.7k 阅读

13 ·

CC 4.0 BY-SA版权

文章标签：

#macos

于 2023-12-26 11:27:45 首次发布

本文介绍了在MacBookPro14寸M1Pro处理器上为特定版本的PyTorch安装torch_sparse、torch_cluster等包的两种方法：一是从官网下载whl文件并使用pip安装，二是通过指定torch版本并依次安装相关依赖。

部署运行你感兴趣的模型镜像

背景

默认电脑已经安装好 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

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理

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

背景

方法一

方法二

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