Commit c2dcc5fd authored by limm's avatar limm
Browse files

optimize README.md

parent f8f3b26f
# <div align="center"><strong>PyTorch Cluster</strong></div> # <div align="center"><strong>PyTorch Cluster</strong></div>
## 简介 ## 简介
PyTorch Cluster一个高度优化的图聚类算法的小型扩展库组成,用于PyTorch。PyTorch Cluster官方github地址:[https://github.com/rusty1s/pytorch_cluster](https://github.com/rusty1s/pytorch_cluster) PyTorch Cluster一个小型的扩展库,其中包含了高度优化的图聚类算法,用于PyTorch中使用。该包包括以下聚类算法:Graclus,Voxel Grid Pooling,迭代最远点采样,k-NN和Radius图生成,基于最近点的聚类,随机游走采样等。DAS软件栈中的PyTorch Cluster版本,不仅保证了组件核心功能在DCU加速卡的可用性,还针对DCU特有的硬件架构进行了深度定制优化。这使得开发者能够以极低的成本,轻松实现应用程序在DCU加速卡上的快速迁移和性能提升。目前支持Pytorch1.13 Pyotrch2.1 Pytorch2.3
## 安装 ## 安装
...@@ -31,7 +31,7 @@ pip install torch* (下载的torch的whl包) ...@@ -31,7 +31,7 @@ pip install torch* (下载的torch的whl包)
#### 源码编译安装 #### 源码编译安装
```shell ```shell
git clone -b 1.6.0-release http://developer.hpccube.com/codes/aicomponent/torch-cluster.git git clone -b 1.6.0-release http://developer.hpccube.com/codes/aicomponent/torch-cluster.git
python pymap_script.py /path/to/pytorch_cluster export FORCE_CUDA=1 or python pymap_script.py /path/to/pytorch_cluster(二选一,该操作是把HIP(CUDA)代码编译进去)
cd pytorch_cluster cd pytorch_cluster
python setup.py bdist_wheel python setup.py bdist_wheel
pip install dist/*.whl pip install dist/*.whl
...@@ -52,4 +52,4 @@ ln -s /usr/local/lib/python3.8/site-packages/torch_cluster/* . ...@@ -52,4 +52,4 @@ ln -s /usr/local/lib/python3.8/site-packages/torch_cluster/* .
## 参考资料 ## 参考资料
https://github.com/rusty1s/pytorch_cluster PyTorch Cluster官方github地址:[https://github.com/rusty1s/pytorch_cluster](https://github.com/rusty1s/pytorch_cluster)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment