python3.8安装torch、torch-sparse、torch-scatter等库

python3.8安装torch、torch-sparse、torch-scatter等库

  • 总体配置

到torch-sparse这四个库的作者git上查询,发现作者给出了不同版本需要的pytorch版本,折中选择了说明为 “这个版本只有在pytorch-1.6.0下才可使用” 的那一版。

总体配置

python3.8+cuda10.1
torch1.6.0+cu101
torch-cluster1.5.7
torch-geometric1.6.1
torch-scatter 2.0.5
torchvision0.7.0

#这个配置在一般电脑上亲测好用,如果电脑显卡较高,要求cuda版本会高,那就要重新进行版本匹配。

有两种下载方法:
1、直接下载whl文件到本地,进入文件所在目录pip整个文件名
2、到pytorch官网/git下载所有版本的torch/torch-cluster等

whl文件下载地址:
torch:https://download.pytorch.org/whl/torch_stable.html
四个子库:https://pytorch-geometric.com/whl/torch-1.6.0.html
(注意:这里把torch-1.6.0替换成torch1.4.0或1.7.0都可以下载对应版本的库)

你可能感兴趣的:(pytorch)