1、Numpy快速入门
https://ptorch.com/docs/9/numpy-quickstart-tutorial
2、PyTorch中文文档
3、PyTorch聚类算法实现及其优化的扩展库(pytorch_cluster)
PyTorch Cluster:转载https://ptorch.com/news/131.html
该软件包包含一个用于PyTorch
的高度优化图形集群算法的小型扩展库。所有包含的操作都适用于不同的数据类型,并针对CPU和GPU实施。
安装
检查nvcc
是否可以从终端访问,例如nvcc --version
。如果没有,请将cuda(/usr/local/cuda/bin)
添加到您的$PATH
变量。然后运行:
pip install cffi torch-cluster
python setup.py install
运行测试
python setup.py test