• 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

  • 相关阅读:
    POJ 2778 DNA Sequence(AC自动机+矩阵)
    Challenge & Growth —— 从这里开始
    京东云
    [Done] Codeforces Round #562 (Div. 2) 题解
    HDU 3486 Interviewe
    Codeforces Round #529 (Div. 3) 题解
    Wannafly 挑战赛 19 参考题解
    第十六届上海大学程序设计联赛春季赛暨上海高校金马五校赛 题解
    2018年长沙理工大学第十三届程序设计竞赛 题解
    POJ 3017 Cut the Sequence
  • 原文地址:https://www.cnblogs.com/liliwang/p/14213646.html
Copyright © 2020-2023  润新知