Commit adfff486 authored by limm's avatar limm
Browse files

optimize README.md

parent 7c7357d5
# <div aligh="center"><strong>PyTorch Sparse</strong></div> # <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方式安装 ### 使用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版本的安装包 (如果为空需要自己按照下面步骤编译) 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 ...@@ -59,6 +59,4 @@ static auto registry = torch::RegisterOperators().op("torch_sparse::cuda_version
代码已修改,目前已经支持torch2.1, 如果编译torch1.13 需要把上面代码修改回去. 代码已修改,目前已经支持torch2.1, 如果编译torch1.13 需要把上面代码修改回去.
## 参考资料 ## 参考资料
```shell [https://github.com/rusty1s/pytorch_sparse](https://github.com/rusty1s/pytorch_sparse)
https://github.com/rusty1s/pytorch_sparse
```
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