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
25e55f3f
"src/libtorio/ffmpeg/pybind/pybind.cpp" did not exist on "0688863cec57675c42db6802c85056fd2bcaaf0c"
Commit
25e55f3f
authored
Aug 02, 2024
by
limm
Browse files
fix README.md
parent
19fd8251
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
11 deletions
+10
-11
README.md
README.md
+10
-11
No files found.
README.md
View file @
25e55f3f
...
@@ -3,7 +3,7 @@
...
@@ -3,7 +3,7 @@
Pytorch Sparce 是 SplineCNN 的基于样条的卷积算子
Pytorch Sparce 是 SplineCNN 的基于样条的卷积算子
### 使用pip方式安装
### 使用pip方式安装
torch-sparce whl包下载目录:
[
http://10.6.10.68:8000/customized/torch-sparse/dtk2
310
/
](
http://10.6.10.68:8000/customized/torch-sparse/dtk2
310
/
)
. 目前只提供有python
3.8
版本的安装包
torch-sparce whl包下载目录:
[
http://10.6.10.68:8000/customized/torch-sparse/dtk2
4.04
/
](
http://10.6.10.68:8000/customized/torch-sparse/dtk2
4.04
/
)
. 目前只提供有python
.10
版本的安装包
```
shell
```
shell
pip
install
torch_sparse
*
(
下载的torch_sparse的whl包
)
pip
install
torch_sparse
*
(
下载的torch_sparse的whl包
)
```
```
...
@@ -15,25 +15,24 @@ pip install torch_sparse* (下载的torch_sparse的whl包)
...
@@ -15,25 +15,24 @@ pip install torch_sparse* (下载的torch_sparse的whl包)
pip
install
numpy
pip
install
numpy
pip
install
'urllib3==1.26.14'
pip
install
'urllib3==1.26.14'
pip
install
pytest
pip
install
pytest
pip insta
;;
wheel
pip
insta
ll
wheel
```
```
-
在首页 | 光合开发者社区下载 det2
3.10
解压在 /opt/ 路径下,并建立软连接,例如:
-
在首页 | 光合开发者社区下载 det2
4.04
解压在 /opt/ 路径下,并建立软连接,例如:
```
shell
```
shell
cd
/opt
cd
/opt
wget http://10.6.10.68:8000/dtk-
release/dtk23.10
/CentOS7.6/DTK-2
3.10
-CentOS7.6-x86_64.tar.gz
wget http://10.6.10.68:8000/dtk-
pkg/dtk24.04/release
/CentOS7.6/DTK-2
4.04
-CentOS7.6-x86_64.tar.gz
tar
-zxvf
DTK-2
3.10
-CentOS7.6-x86_64.tar.gz
tar
-zxvf
DTK-2
4.04
-CentOS7.6-x86_64.tar.gz
ln
-s
dtk-2
3.10
dtk
ln
-s
dtk-2
4.04
dtk
source
/opt/dtk/env.sh
source
/opt/dtk/env.sh
```
```
-
安装pytorch. pytorch whl包下载目录:
[
http://10.6.10.68:8000/debug/pytorch/dtk2
3.10/hipify
](
http://10.6.10.68:8000/debug/pytorch/dtk23.10/hipify
)
. 根据需求下载对应的版本,安装如下:
-
安装pytorch. pytorch whl包下载目录:
[
http://10.6.10.68:8000/debug/pytorch/dtk2
4.04/torch-2.1.0%2Bgit1c51f6a.abi1.dtk2404-cp310-cp310-linux_x86_64.whl
](
http://10.6.10.68:8000/debug/pytorch/dtk24.04/torch-2.1.0%2Bgit1c51f6a.abi1.dtk2404-cp310-cp310-linux_x86_64.whl
)
. 根据需求下载对应的版本,安装如下:
```
shell
```
shell
pip
install
torch
*
(
下载的torch的whl包
)
pip
install
torch
*
(
下载的torch的whl包
)
```
```
#### 源码下载编译安装
#### 源码下载编译安装
```
shell
```
shell
git clone
-b
0.6.15-release http://developer.hpccube.com/codes/aicomponent/torch-sparce.git
git clone
-b
0.6.16-release http://developer.hpccube.com/codes/aicomponent/torch-sparce.git
python pymap_script.py /path/to/torch-sparce
cd
torch-sparce
cd
torch-sparce
python setup.py bdist_wheel
python setup.py bdist_wheel
pip
install
dist/
*
.whl
pip
install
dist/
*
.whl
...
@@ -50,10 +49,10 @@ pytest
...
@@ -50,10 +49,10 @@ pytest
```
```
find / -name "_version_cpu.so"
find / -name "_version_cpu.so"
cd /torch-sparce/torch_sparse
cd /torch-sparce/torch_sparse
ln -s /usr/local/lib/python3.
8
/site-packages/torch_sparse/* .
ln -s /usr/local/lib/python3.
10
/site-packages/torch_sparse/* .
```
```
-
编译torch_sparse==0.6.1
5
目前不支持torch2.1版本的torch,如果编译torch2.1版本的torch_sparse0.6.15版本需要修改一些代码,文件路径pytorch_sparse/csrc/version.cpp
-
编译torch_sparse==0.6.1
6
目前不支持torch2.1版本的torch,如果编译torch2.1版本的torch_sparse0.6.15版本需要修改一些代码,文件路径pytorch_sparse/csrc/version.cpp
```
shell
```
shell
static auto registry
=
torch::RegisterOperators
()
.op
(
"torch_sparse::cuda_version"
, &sparse::cuda_version
)
;
-> static auto registry
=
torch::RegisterOperators
()
.op
(
"torch_sparse::cuda_version"
,
[]
{
return
sparse::cuda_version
()
;
})
;
static auto registry
=
torch::RegisterOperators
()
.op
(
"torch_sparse::cuda_version"
, &sparse::cuda_version
)
;
-> static auto registry
=
torch::RegisterOperators
()
.op
(
"torch_sparse::cuda_version"
,
[]
{
return
sparse::cuda_version
()
;
})
;
```
```
...
...
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