【环境配置】torch-scatter/cluster/sparse等的版本匹配

本文提供了针对不同Python版本(3.6, 3.7, 3.8)的PyTorch及其相关生态组件(如torch-scatter, torch-cluster等)的具体版本配置方案。这些配置适用于进行图神经网络研究和开发。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

以下是几个可用的版本配置:

python=3.8​​​​​​​

           torch==1.8.0+cu111
           torch-scatter==2.0.6
           torch-cluster==1.5.9
           torch-sparse==0.6.9
           torch-geometric==1.6.3

python=3.7

           torch==1.1.0
           torch-scatter==1.3.2
           torch-cluster==1.4.5
           torch-sparse==0.4.3
           torch-spline-conv==1.1.1
           torch-geometric==1.3.2
           torchvision==0.3.0

python==3.6

     torch==1.3.0
     torch-scatter==1.3.2
     torch-cluster==1.4.5
     torch-sparse==0.4.3
     torch-spline-conv==1.1.1
     torch-geometric==1.3.2
     torchvision==0.4.1


致谢:https://blog.youkuaiyun.com/u014714362/article/details/108614868

### 安装 `torch-geometric` 和其依赖项 (`torch-scatter`, `torch-sparse`, `torch-cluster`),建议遵循官方推荐的方法来设置环境。这通常涉及使用特定于 PyTorch 版本和 CUDA 配置的预编译二进制文件。 #### 方法一:通过 PyPI 安装最新稳定版 对于大多数用户而言,最简单的方式是从 Python Package Index (PyPI) 获取最新的稳定版本: ```bash pip install torch_geometric[full] ``` 这条命令会自动处理所有必要的依赖关系并安装它们[^1]。 #### 方法二:针对特定 PyTorch/CUDA 组合的手动安装 如果需要匹配某个具体的 PyTorch 或者 CUDA 版本,则应该先确认已正确安装所需版本的 PyTorch 后再继续如下操作: 假设已经安装了 PyTorch 1.7.1 并且想要兼容 CUDA 11.0 的情况下,可以从指定 URL 下载适合当前系统的 wheel 文件,并依次执行下面几条指令完成各个组件的部署: ```bash pip install https://data.pyg.org/whl/torch-1.5.8-cp37-cp37m-linux_x86_64.whl pip install https://data.pyg.org/whl/torch-1.7.1%2Bcu110/torch_scatter-2.0.5-cp37-cp37m-linux_x86_64.whl pip install https://data.pyg.org/whl/torch-1.7.1%2Bcu110/torch_sparse-0.6.8-cp37-cp37m-linux_x86_64.whl pip install https://data.pyg.org/whl/torch-1.7.1%2Bcu110/torch_spline_conv-1.2.0-cp37-cp37m-linux_x86_64.whl pip install torch_geometric==1.7.0 ``` 请注意替换上述链接中的版本号以适应实际使用的 PyTorch 和操作系统架构[^2]。 #### 检查安装成功与否 可以通过导入这些库来进行简单的测试,验证是否安装无误: ```python import torch from torch_geometric.data import Data print(torch.__version__) print(Data) ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值