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

         即可!

猜你喜欢

转载自blog.csdn.net/baidu_30506559/article/details/121915734