Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
torch-cluster
Commits
c2dcc5fd
Commit
c2dcc5fd
authored
Sep 09, 2024
by
limm
Browse files
optimize README.md
parent
f8f3b26f
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
README.md
README.md
+3
-3
No files found.
README.md
View file @
c2dcc5fd
# <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/pyt
orch
_c
luster
](
https://github.com/rusty1s/pytorch_cluster
)
PyTorch Cluster
是
一个
小型的扩展库,其中包含了
高度优化的图聚类算法,用于
在
PyTorch
中使用。该包包括以下聚类算法:Graclus,Voxel Grid Pooling,迭代最远点采样,k-NN和Radius图生成,基于最近点的聚类,随机游走采样等。DAS软件栈中的PyT
orch
C
luster
版本,不仅保证了组件核心功能在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
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment