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
7c7357d5
"torchvision/vscode:/vscode.git/clone" did not exist on "054432d2d00d80746f2eb55d959a92c2ea3858e3"
Commit
7c7357d5
authored
Aug 06, 2024
by
limm
Browse files
modify the README.md
parent
25e55f3f
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
3 deletions
+4
-3
README.md
README.md
+4
-3
No files found.
README.md
View file @
7c7357d5
...
...
@@ -3,7 +3,7 @@
Pytorch Sparce 是 SplineCNN 的基于样条的卷积算子
### 使用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版本的安装包
(如果为空需要自己按照下面步骤编译)
```
shell
pip
install
torch_sparse
*
(
下载的torch_sparse的whl包
)
```
...
...
@@ -26,7 +26,7 @@ ln -s dtk-24.04 dtk
source
/opt/dtk/env.sh
```
-
安装pytorch. pytorch whl包下载目录:
[
http://10.6.10.68:8000/debug/pytorch/dtk24.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
)
. 根据需求下载对应的版本,安装如下:
-
安装pytorch. pytorch whl包下载目录:
[
http://10.6.10.68:8000/debug/pytorch/dtk24.04/
](
http://10.6.10.68:8000/debug/pytorch/dtk24.04/
)
. 根据需求下载对应
系统下
的版本,安装如下:
```
shell
pip
install
torch
*
(
下载的torch的whl包
)
```
...
...
@@ -52,10 +52,11 @@ cd /torch-sparce/torch_sparse
ln -s /usr/local/lib/python3.10/site-packages/torch_sparse/* .
```
-
编译torch_sparse==0.6.16 目前不支持torch2.1版本的torch,如果编译torch2.1版本的torch_sparse0.6.1
5
版本需要修改一些代码,文件路径pytorch_sparse/csrc/version.cpp
-
编译torch_sparse==0.6.16 目前不支持torch2.1版本的torch,如果编译torch2.1版本的torch_sparse0.6.1
6
版本需要修改一些代码,文件路径pytorch_sparse/csrc/version.cpp
```
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
()
;
})
;
```
代码已修改,目前已经支持torch2.1, 如果编译torch1.13 需要把上面代码修改回去.
## 参考资料
```
shell
...
...
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