当前位置:   article > 正文

python3.8安装torch、torch-sparse、torch-scatter等库_python3.8有torch包嘛

python3.8有torch包嘛

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都可以下载对应版本的库)

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/203514?site
推荐阅读
相关标签
  

闽ICP备14008679号