根据自己电脑的配置,选择相应的版本,建议用pip安装
如果觉得这样直接安装慢,也可以根据给出的网址下载.whl文件进行半自动安装
使用官网提供的安装教程,下面是torch1.6+cuda10.1+pytorch-geometric的安装教程
pip install torch-scatter==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-sparse==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-cluster==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-spline-conv==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-geometric
注意:pytorch版本前后要一致。
pytorch-cpu=0.4.0下载地址
pytorch下载地址
conda create -n pytorch1 python=3.6
activate pytorch
比如我下载到了D:\Chrome下载文件目录下
D:
cd D:\Chrome下载
pip install torch-0.4.0-cp36-cp36m-win_amd64.whl
importError :DLL load failed:找不到指定的模块。
解决办法:conda install numpy
下载地址