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
9d1009f3
Commit
9d1009f3
authored
May 10, 2023
by
quyuanhao123
Browse files
Update README_HIP.md
parent
f1c34c51
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
4 deletions
+5
-4
README_HIP.md
README_HIP.md
+5
-4
No files found.
README_HIP.md
View file @
9d1009f3
**编译安装步骤**
-
1. 安装和选用dtk版本对应的pytorch torchvision whl(可在光合社区下载https://developer.hpccube.com/tool/)
例如:
pip install https://cancon.hpccube.com:65024/directlink/4/pytorch/dtk22.10/torch-1.10.0a0+git2040069.dtk2210-cp38-cp38-manylinux2014_x86_64.whl
pip install https://cancon.hpccube.com:65024/directlink/4/vision/dtk22.10/torchvision-0.10.0a0+e04d001.dtk2210-cp38-cp38-manylinux2014_x86_64.whl
例如:
pip install https://cancon.hpccube.com:65024/directlink/4/pytorch/dtk22.10/torch-1.10.0a0+git2040069.dtk2210-cp38-cp38-manylinux2014_x86_64.whl
pip install https://cancon.hpccube.com:65024/directlink/4/vision/dtk22.10/torchvision-0.10.0a0+e04d001.dtk2210-cp38-cp38-manylinux2014_x86_64.whl
-
2. 自定义编译时的环境变量
激活对应的conda环境:
...
...
@@ -45,6 +46,6 @@ pip install https://cancon.hpccube.com:65024/directlink/4/vision/dtk22.10/torchv
export CXX=hipcc
-
[ ]
3. 编译安装
-
3. 编译安装
python setup.py install
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