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

  • 相关阅读:
    iOS socket编程 第三方库 AsyncSocket(GCDAsyncSocket)
    JS中reduce方法
    程序员的运动建议
    Vuex(三)—— getters,mapGetters,...mapGetters详解
    微信小程序之使用函数防抖与函数节流
    JS 异步(callback→Promise→async/await)
    圈子与网络
    社会经验4
    社会经验3
    爱情(。_。)大忌
  • 原文地址:https://www.cnblogs.com/yqpy/p/13498200.html
Copyright © 2011-2022 走看看