PyTorch聚类算法实现及其优化的扩展库(pytorch_cluster)

Song • 1466 次浏览 • 0 个回复 • 2018年02月26日

PyTorch Cluster

该软件包包含一个用于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

原创文章,转载请注明 :PyTorch聚类算法实现及其优化的扩展库(pytorch_cluster) - pytorch中文网
原文出处: https://ptorch.com/news/131.html
问题交流群 :168117787
提交评论
要回复文章请先登录注册
用户评论
  • 没有评论
Pytorch是什么?关于Pytorch! 关于PyTorch的教程,论文,项目,社区等的精选清单。(the-incredible-pytorch)