Commit 7ab1b528 authored by panning's avatar panning
Browse files

add README_HIP.md

parent 5e8e2f12
## **安装**
### **安装方式**
#### **使用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)下载 DTK-23.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.14.1 http://developer.hpccube.com/codes/aicomponent/vision.git
cd vision
PYTORCH_ROCM_ARCH="gfx906;gfx926" python3 setup.py bdist_wheel
pip3 install dist/vision*
```
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