# MMCV

## 安装
### mmcv支持
+ python3.7
+ python3.8
+ python3.9

### 使用pip安装
fastfold whl包下载目录：[https://cancon.hpccube.com:65024/4/main/mmcv/dtk23.04](https://cancon.hpccube.com:65024/4/main/mmcv/dtk23.04)
选择对应的pytorch版本和python版本下载对应mmcv的whl包
```shell
pip install mmcv* (下载的mmcv的whl包)
```
### 使用源码安装

#### 编译环境准备
pytorch whl包下载目录：[https://cancon.hpccube.com:65024/4/main/pytorch/dtk23.04](https://cancon.hpccube.com:65024/4/main/pytorch/dtk23.04)
根据python版本,下载对应pytorch的whl包
```shell
pip install torch* (下载的torch的whl包)
```

```shell
pip install setuptools=59.5.0 wheel
```

#### 编译安装
```shell
git clone -b dtk-23.04_v1.6.1 https://developer.hpccube.com/codes/aicomponent/mmcv
cd mmcv
export MMCV_BUILD_VERSION=abix.dtkxxx
MMCV_WITH_OPS=1 ROCM_HOME=${ROCM_PATH} python3 setup.py -v bdist_wheel
pip install dist/mmcv*
```

## Note
+ 若使用 pip install 下载安装过慢，可添加源：-i https://pypi.tuna.tsinghua.edu.cn/simple/
+ MMCV_BUILD_VERSION为编译的版本号设置，版本号为1.6.1+gitxxx.abix.dtkxxx gitxxx：为代码自动获取；abi0：使用devtools的gcc编译；abi1：使用非devtools的gcc编译; dtkxxx为dtk的版本号：例：dtk2304
+ ROCM_PATH为dtk的路径，默认为/opt/dtkxxx