torch-scatter -cluster -spline -geometric 安装

今天真的是吐血,一定要把这个分享出去,让更多的人少走弯路。很少有人分享torch-scatter -cluster -spline -geometric 安装,找淘宝店给我安装,他竟然说不会!我只能自己寻找方法了!

首先,相信很多人看过这个教程:https://github.com/rusty1s/pytorch_geometric#pytorch-170

我来排一排这个教程的坑,就可以安装成功了。

一、环境变量切换

一定要切换到想要安装库的环境变量中

conda env list          # 查看变量列表

activate 变量名        # 激活要安装库的变量

二、pytorch版本的选择

如果打开下载地址的上一级目录:https://pytorch-geometric.com/whl/,可以看到pytorch版本(注:没有1.7.1或1.6.**等版本的安装链接,如果已安装的pytorch是1.7.**版本,安装链接和1.7.0一样

三、cuda的选择

与电脑已安装版本对应,查看版本后,再选择安装链接。

另,以cu101为例,教程中 “${CUDA}” 改为 ”cu101“,

我确定Cuda版本的方法是:用everything软件搜索“cuda”关键字,在anaconda安装目录下发现如下图所示文件夹,从而确定cuda版本

 

题外话:

everything是一个非常方便且高效的文件搜索小工具,安装包可自取

链接:https://pan.baidu.com/s/18r4NGTTbchrKvBCM7bjxKA 
提取码:vqja 

四、安装的完成命令

则完整命令为(以Python3.7、pytorch1.7.1、cuda101为例):

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu101.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+cu101.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+cu101.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+cu101.html
pip install torch-geometric

五、安装完成测试

在cmd中,进入python环境:

(py37) C:\Users\xiaozhe>python
Python 3.7.9 (default, Aug 31 2020, 17:10:11) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.

 倒入库(注:库的名字中“-”改为下划线“_"才能成功导入):

>>> import torch_scatter
>>> import torch_sparse
>>> import torch_cluster
>>> import torch_spline_conv
>>> import torch_geometric

不报错就成功了 !

  • 6
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值