README.md

## **安装**

### **安装方式**

#### **使用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.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*
```