PyTorch Geometric Temporal 环境安装 (亲测有效)

1. PyTorch Geometric Temporal 介绍

PyTorch Geometric Temporal 是 PyTorch Geometric 的一个时间(动态)扩展库。

该图书馆包括各种动态和时间几何深度学习,嵌入和时空回归方法从各种已发表的研究论文。此外,它还附带了一个易于使用的数据集加载器、列车测试分离器和用于动态和时态图的时态快速迭代器。该框架自然地提供了 GPU 支持。它还提供了大量的基准数据集,包括人口统计学预测、共享经济、能源生产和网络流量管理等领域。最后,您还可以创建自己的数据集。

该软件包与 Pytorch Lightning 接口良好,允许在 cpu 上进行培训,单个和多个 gpu 即开即用。看看这个使用 PyTorch Geometric Temporal 和 PyTorch lightning 的介绍性示例。

2. Github 地址:GitHub - benedekrozemberczki/pytorch_geometric_temporal: PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

下载ZIP:

3. 要求Python版本=3.7!!(没有的话,新建一个虚拟环境)

conda create -n pytorch_geo Python=3.7 

4. 确定自己的电脑现安装版本(GPU版本)

 1. 是否有GPU,是否安装CUDA ,是否安装了Pytorch?

 2. CUDA版本是多少?查看 CUDA 版本 正确方法(亲测有效)

     Pytorch版本为多少?(直接conda list 查看即可)

     当然,如果没有安装了Pytorch,建议先匹配自己的电脑先安装一个。

      Pytorch 安装 GPU版本

5. 解压zip,查看docs/requirements.txt

networkx
numpy
https://download.pytorch.org/whl/cpu/torch-1.5.0%2Bcpu-cp37-cp37m-linux_x86_64.whl
https://s3.eu-central-1.amazonaws.com/pytorch-geometric.com/whl/torch-1.5.0/torch_scatter-latest%2Bcpu-cp37-cp37m-linux_x86_64.whl
https://s3.eu-central-1.amazonaws.com/pytorch-geometric.com/whl/torch-1.5.0/torch_sparse-latest%2Bcpu-cp37-cp37m-linux_x86_64.whl
https://s3.eu-central-1.amazonaws.com/pytorch-geometric.com/whl/torch-1.5.0/torch_cluster-latest%2Bcpu-cp37-cp37m-linux_x86_64.whl
https://s3.eu-central-1.amazonaws.com/pytorch-geometric.com/whl/torch-1.5.0/torch_spline_conv-latest%2Bcpu-cp37-cp37m-linux_x86_64.whl
torch_geometric
sphinx==4.0.2
sphinx_rtd_theme==0.5.2
six

————重要!!!————

        1. 我们要根据自己的情况更改requirements.txt:

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

        在这个链接中找到你所需要的版本。

        2. 比如,我的电脑是window系统,已经安装了CUDA=10.2 的Pytorch=1.7.0,那么就可以将requirements.txt改成:

networkx
numpy
https://pytorch-geometric.com/whl/torch-1.7.0/torch_scatter-latest+cu102-cp37-cp37m-win_amd64.whl
https://pytorch-geometric.com/whl/torch-1.7.0/torch_sparse-latest+cu102-cp37-cp37m-win_amd64.whl
https://pytorch-geometric.com/whl/torch-1.7.0/torch_cluster-latest+cu102-cp37-cp37m-win_amd64.whl
https://pytorch-geometric.com/whl/torch-1.7.0/torch_spline_conv-latest+cu102-cp37-cp37m-win_amd64.whl
torch_geometric
sphinx==4.0.2
sphinx_rtd_theme==0.5.2
six

        其他的环境和版本可以照葫芦画瓢。

6. 安装requirements.txt内的包

        1. 打开anaconda prompt,激活之前新建的虚拟环境pytorch_geo 

        2. 切换路径至pytorch_geometric_temporal-master\docs

cd  D:\pytorch_geometric_temporal-master\docs

        3.  输入:

pip install -r requirements.txt

         即可!

  • 2
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值