本文主要是介绍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
即可!
这篇关于PyTorch Geometric Temporal 环境安装 (亲测有效)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!