PyTorch-Geometric安装
本文借鉴了知乎的回答
问题:
在如下环境中:
- 操作系统:Win10
- Cuda版本:Cuda 10.2
- Pytorch版本:Pytorch 1.6
- Python版本:Python 3.7
已经安装好pytorch的GPU版本,要安装pytorch-geometric。
步骤:
若想安装pytorch-geometric,需要安装以下包:
- torch-scatter
- torch-sparse
- torch-cluster
- torch-spline-conv
- torch-geometric
第5个包可以直接用pip安装。
前4个包直接用pip安装会报错。
解决方法:
直接用whl文件进行安装。
该项目提供了各个版本的whl文件。
并且介绍了各个版本的whl文件使用方法。
如果要总结一下作者在项目中介绍的方法,简而言之就是如下命令:
pip install ${PackageName} == latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.6.0.html
其中${PackageName} 是前四个包的名称,即torch-scatter,torch-sparse,torch-cluster,torch-spline-conv;${CUDA} 是CUDA的版本号,可以从cpu,cu92,cu101,cu102中选择。