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
vision
Commits
25729a99
Commit
25729a99
authored
May 09, 2023
by
panning
Browse files
add README_HIP.md
parent
e04d001a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
54 additions
and
0 deletions
+54
-0
README_HIP.md
README_HIP.md
+54
-0
No files found.
README_HIP.md
0 → 100644
View file @
25729a99
## **安装**
### **安装方式**
#### **使用pip方式安装**
vision whl包下载位置:
[
vision | 光合开发者社区
](
https://cancon.hpccube.com:65024/4/main/vision
)
,进入对应DTK目录(dtk23.04)根据Python版本选择安装包。
```
bash
pip3
install
torchvision
*
```
#### **使用源码安装**
##### 编译环境准备
-
安装相关依赖
```
#可通过conda或源码编译安装libpng/libjpeg
conda install libpng
conda install jpeg
#注:libpng和libjpeg必须在编译时可用,确保它在标准库位置可用,否则,分别在环境变量TORCHVISION_INCLUDE和TORCHVISION_LIBRARY中添加头文件路径和库路径。
pip3 install pybind11
pip3 install 'numpy<=1.23.5'
pip3 install 'urllib3==1.26.14'
pip3 install requests
pip3 install wheel
```
-
在
[
首页 | 光合开发者社区
](
https://cancon.hpccube.com:65024/1/main
)
下载 dtk23.04 解压至 /opt/ 路径下,并建立软链接
```
cd /opt && ln -s dtk-23.04 dtk
source /opt/dtk/env.sh
```
-
在
[
pytorch | 光合开发者社区
](
https://cancon.hpccube.com:65024/4/main/pytorch
)
下载对应的 pytorch安装包(需对应dtk23.04版本与 Python版本)
```
pip3 install torch*
```
##### 编译安装
```
bash
git clone
-b
dtk-23.04-v0.10.0 http://developer.hpccube.com/codes/aicomponent/vision.git
cd
vision
PYTORCH_ROCM_ARCH
=
"gfx906;gfx926"
python3 setup.py bdist_wheel
pip3
install
dist/vision
*
```
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