zoukankan      html  css  js  c++  java
  • torch_geometric安装

    因为需要跑一些别人的GCN的任务,所以就要安装相关的包;

    首先要注意cuda版本和pytorch中的对应关系,必须严格对应,比如我的机器上,cuda版本是10.1;

     pytorch的cuda版本也是10.1,之前其实是10.2,又重装了一遍;

    pytorch版本是1.4; python版本是3.6;然后在这个网页上

    https://pytorch-geometric.com/whl/torch-1.4.0.html

    CUDA=cu101
    pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
    pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
    pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html
    pip install torch-cluster==1.4.5
    pip install torch-geometric

    可以先把各个包按照要求都下下来,然后直接pip install XX就很快了;

    这里有个问题需要注意一下,安装torch_geometric,其实默认会安装最新版本(目前是1.6.1),但有些方法在1.4.3之后就不支持了。

    比如torch_geometric.utils.scatter_这个方法,在1.5及以后的版本就不支持了,所以代码中有用到这个方法的,要不就另写一个函数代替,要不就把版本退到1.4.3

  • 相关阅读:
    沈阳集训day2
    ac自动机
    2018沈阳集训day1
    洛谷P1875 佳佳的魔法药水
    洛谷P1941 飞扬的小鸟
    Noip2016day2
    1123: [POI2008]BLO
    1718: [Usaco2006 Jan] Redundant Paths 分离的路径
    P3119 [USACO15JAN]草鉴定Grass Cownoisseur
    [LeetCode] Clone Graph
  • 原文地址:https://www.cnblogs.com/yqpy/p/13498200.html
Copyright © 2011-2022 走看看