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-sparse
Commits
adfff486
Commit
adfff486
authored
Sep 09, 2024
by
limm
Browse files
optimize README.md
parent
7c7357d5
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
4 deletions
+2
-4
README.md
README.md
+2
-4
No files found.
README.md
View file @
adfff486
# <div aligh="center"><strong>PyTorch Sparse</strong></div>
## 简介
Pytorch Sparce 是
SplineCNN 的基于样条的卷积算子
Pytorch Sparce 是
个包包含了一个小型扩展库,用于优化支持自动微分的稀疏矩阵操作。目前,这个包包括以下方法:Coalesce,Transpose,Sparse Dense Matrix Multiplication,Sparse Sparse Matrix Multiplication 所有包含的操作都支持不同类型的数据,并且既在CPU上也在GPU上实现了。DAS软件栈中的PyTorch Sparce版本,不仅保证了组件核心功能在DCU加速卡的可用性,还针对DCU特有的硬件架构进行了深度定制优化。这使得开发者能够以极低的成本,轻松实现应用程序在DCU加速卡上的快速迁移和性能提升。目前支持Pytorch1.13 Pyotrch2.1 Pytorch2.3
### 使用pip方式安装
torch-sparce whl包下载目录:
[
http://10.6.10.68:8000/customized/torch-sparse/dtk24.04/
](
http://10.6.10.68:8000/customized/torch-sparse/dtk24.04/
)
. 目前只提供有python.10版本的安装包 (如果为空需要自己按照下面步骤编译)
...
...
@@ -59,6 +59,4 @@ static auto registry = torch::RegisterOperators().op("torch_sparse::cuda_version
代码已修改,目前已经支持torch2.1, 如果编译torch1.13 需要把上面代码修改回去.
## 参考资料
```
shell
https://github.com/rusty1s/pytorch_sparse
```
[
https://github.com/rusty1s/pytorch_sparse
](
https://github.com/rusty1s/pytorch_sparse
)
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