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

  • 相关阅读:
    DA_06_iptables 与 firewalld 防火墙
    DA_05_Linux(CentOS6.7) 安装MySql5.7数据库
    DA_04_解决Xshell中文乱码问题
    3.NumPy
    2.NumPy简介
    1.python环境安装
    4.5. scrapy两大爬虫类_Spider
    redis 锦集
    一位资深程序员大牛给予Java初学者的学习路线建议
    idea 使用过程中的一些设置记录
  • 原文地址:https://www.cnblogs.com/yqpy/p/13498200.html
Copyright © 2011-2022 走看看