1.步骤一
从下面这个链接进去,选择适合自己电脑版本的pytorch。有同学问:不是torch吗,为什么用pytorch?这里,可以自行百度一下torch和pytorch的区别,大概就是pytorch是torch的python版本,功能大概相似,区别不是很大。
pytorch官网:https://pytorch.org/get-started/locally/
我是再anaconda环境下装的,这个比较容易。比如我在环境下的终端运行的是:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
2.步骤二
下载好pytorch之后,从下面这个链接装torch-scatter, torch-sparse, torch-cluster, torch-spline-conv这些库。
pytorch-geometric官网:https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
在anaconda环境下只运行一句就OK,除了通过Anaconda安装,也有通过Wheels,或者从源码安装,或多或少都有点麻烦,比如超时之类的。(一把辛酸泪。。。)
比如我在环境下的终端运行的是:
conda install pyg -c pyg -c conda-forge
END
完成以上步骤,基本就可以用了。当你在查看安装的包的时候,会发现都是带py前缀的,可能是因为都是python版本的吧。这样安装完之后,项目是没有报包的错误了,但又出现了新的错误,还在解决中。。。┭┮﹏┭┮
pytorch 1.10.0 py3.8_cpu_0 pytorch
pytorch-cluster 1.5.9 py38_torch_1.10.0_cpu pyg
pytorch-mutex 1.0 cpu pytorch
pytorch-scatter 2.0.9 py38_torch_1.10.0_cpu pyg
pytorch-sparse 0.6.12 py38_torch_1.10.0_cpu pyg
pytorch-spline-conv 1.2.1 py38_torch_1.10.0_cpu pyg
最后,也给上CUDA的下载链接:https://developer.nvidia.com/cuda-toolkit-archive