【Python】安装torch_geometric、以及torch_scatter系列安装

【Python】安装torch_geometric、以及torch_scatter系列安装

文章目录

  • 【Python】安装torch_geometric、以及torch_scatter系列安装
    • 1. 介绍
    • 2. 方法
      • 2.1 第一步:查看CUDA版本
      • 2.2 第二步:安装
      • 2.3 第三步:检测是否安装成功
    • 3. 参考

1. 介绍

首先,我们要知道:

  • 安装torch_geometric,需要同时安装torch-scatter,torch-sparse,torch-cluster,torch-spline-conv等库

因此,你如果只需要torch_scatter的话,就安装它就好了,但是如果要torch_geometric的话,就需要全都下载了。

2. 方法

2.1 第一步:查看CUDA版本

import torch
print(torch.__version__)
print(torch.version.cuda)

注意:

  • Pytorch和CUDA的版本一定要对应上,否则,即便安装了也会出错调用不了,一般情况下,不要安装最新版本Pytorch,因为CUDA和其他相关依赖库。
  • 如果之前安装了其他版本的CUDA或者Pytorch,要卸载干净了再安装,如果没有卸载干净直接安装,系统就会有多个版本,就算用nvcc --version检测出来有安装该软件,也无法正常调用CUDA。

2.2 第二步:安装

## 先安装这四个
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html

## 最后安装torch-geometric
pip install torch-geometric

其中 {TORCH} 替 换 为 P y T o r c h 的 版 本 号 ,{CUDA} 替换为上面查询得到的CUDA版本号。CUDA 版本号格式 (cpu, cu92, cu101, cu102, cu110, cu111)。例如对于PyTorch 2.0.0 and CUDA 11.7,安装命令如下:

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-2.0.0+cu117.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-2.0.0+cu117.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-2.0.0+cu117.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-2.0.0+cu117.html
pip install torch-geometric

2.3 第三步:检测是否安装成功

$ python
Python 3.7.0 (default, Jun 28 2018, 13:15:42)
[GCC 7.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from torch_geometric.data import DataLoader
>>>
  • 没有出现错误提示,即成功。

3. 参考

【1】https://blog.csdn.net/qq_41800917/article/details/120282841

你可能感兴趣的:(#,Python,使用说明,python)