安装说明
由于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
图2
|