zoukankan      html  css  js  c++  java
  • python3.7+pytorch1.7.1+cuda10.2环境安装torch_geometric

    1、官网上查看自己的环境对应的torch_geometric版本

    (1)github网址:https://github.com/rusty1s/pytorch_geometric

    (2)打开网站之后,能在网页上看到  “Installation”,我用到的版本如下图:

    2、安装torch_geometric的命令

    我的电脑配置是cuda10.2,对应的是cu102(如果不是这个环境,可以更改下面命令中加粗的内容),在命令行中运行如下代码即可:

    pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html

    pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html

    pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html

    pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html

    pip install torch-geometric

     3、是否需要其他依赖包?

    在运行上述命令的过程中,我发现系统会自动下载一些包,通过查看原项目,可以看到有一个setup.py文件,打开后可以看到下图,numpy,  tqdm等都是本项目的依赖包。如果之前没有下载过这些包也不用担心,直接运行第二步骤的代码,系统会自动安装需要的包。

    4、安装出现问题,未安装上torch-cluster和torch-scatter

    可以打开网址https://pytorch-geometric.com/whl/torch-1.7.0.html(如果是其他版本的torch,使用该网址查看对应版本https://pytorch-geometric.com/whl/),找自己需要的版本下载之后放在“script"文件夹下,也可以自建文件夹,我的虚拟环境的配置是python3.7.9,pytorch版本是1.7.1,查找到对应的兼容标签(Compatible tags)是”cp37-cp37m-win_amd64“,因此下载了下图所示的版本。

    总结安装步骤:

    • 下载合适的torch-cluster版本,并放在“script"文件夹下
    • cd到该“script"文件夹
    • 输入命令pip install torch_cluster-1.5.8+cpu-cp37-cp37m-win_amd64.whl,并回车
    • 显示Successfully......       安装成功!!!

    注意:查找到对应的兼容标签的方法:参考https://blog.csdn.net/happywlg123/article/details/107281936

      我们可以通过下面这个经过测试有效的命令,来查看当前Python版本匹配那些文件。
      有效命令如下(直接在命令行中输入,不要在Python解释器输入哦):

          pip debug --verbose

      结果很长,但是我们在这里需要的是”Compatible tags"后面的内容,如下,红圈中就是我们需要的版本。

    一定要注意版本的匹配,如果使用cu102版本的torch-geometric,就要使用cu102版本的torch-cluster和torch-scatter

  • 相关阅读:
    oracle查询哪些sp修改了某些表
    asp.net mvc
    更新计算机驱动
    instr函数的用法
    UNION ALL UNION
    Python机器学习ch02 代码学习2
    Python机器学习 ch02代码学习1
    转载Python切片(小知识点)
    FMCW部分资料连接
    Python基础25 异常堆栈跟踪,释放资源,自定义异常和主动抛出
  • 原文地址:https://www.cnblogs.com/liliwang/p/14213646.html
Copyright © 2011-2022 走看看