安装torch-geometric及其相关库

安装说明

由于torch-geometric依赖的对各种库(包括Torch、torch-scatter、torch-sparse等)和CUDA有版本对应有要求,按照官网的代https://pytorch-geometric.com/whl/网址依次进入找到匹配文件下载安装,进入之后如图1所示。
首先,确认自己电脑安装的CUDA和Torch版本,然后依次安装。以下是我的安装配置流程。

1、下载安装torch_scatter-2.0.8

whl文件下载
https://data.pyg.org/whl/torch-1.8.0%2Bcu111/torch_scatter-2.0.8-cp38-cp38-linux_x86_64.whl
进入本地文件夹找到.whl文件,安装torch_scatter

pip install torch_scatter-2.0.8-cp38-cp38-linux_x86_64.whl

2、安装torch_sparse-0.6.9

whl文件下载
https://data.pyg.org/whl/torch-1.8.0%2Bcu111/torch_sparse-0.6.9-cp38-cp38-linux_x86_64.whl
本地安装torch_sparse

pip install torch_sparse-0.6.9-cp38-cp38-linux_x86_64.whl

3、安装torch-geometric

直接输入下面代码安装

pip install torch-geometric

图1
安装torch-geometric及其相关库_第1张图片

图2
安装torch-geometric及其相关库_第2张图片

你可能感兴趣的:(神经网络,深度学习,pytorch)