Commit 1d3c935b authored by xiabo's avatar xiabo
Browse files

readme修改

parent c7118e58
# MMCV
<div align="center">
<img src="https://raw.githubusercontent.com/open-mmlab/mmcv/master/docs/en/mmcv-logo.png" width="300"/>
</div>
## 安装
### mmcv支持
......@@ -19,19 +21,32 @@ pytorch whl包下载目录:[https://cancon.hpccube.com:65024/4/main/pytorch/dt
根据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
git clone https://developer.hpccube.com/codes/aicomponent/mmcv // 根据编译需要切换分支
cd mmcv
// 编译whl包
MMCV_WITH_OPS=1 ROCM_HOME=${ROCM_PATH} python3 setup.py -v bdist_wheel
pip install dist/mmcv*
// 直接编译安装
MMCV_WITH_OPS=1 ROCM_HOME=${ROCM_PATH} python3 setup.py install
```
## 安装包命名
例:mmcv-2.0.0+gitc7118e5.abi0.dtk2304.torch1.13-cp37-cp37m-manylinux2014_x86_64.whl
- mmcv: 软件名称;
- 2.0.0: 软件版本号;
- gitc7118e5: git号;
- abi0: 对应centos和rocky多系统的标识,取值为abi0(centos7,对应devtools),abi1(rocky8, 非devtools),可使用命令行动态查询获取辅助判断(echo '#include <string>' | gcc -x c++ -E -dM - | fgrep _GLIBCXX_USE_CXX11_ABI),根据各包情况是否兼容多系统;
- dtk2304: 对应dtk大版本号,命令行动态查询取值(/opt/dtk-22.10.1/.info/rocm_version);
- torch1.13: 对应构建的torch版本号,命令行动态查询取值(torch.\_\_version__);
- cp37-cp37m: 对应python版本号;
- linux_x86_64: 系统架构;
## 版本号查询
- \_\_version__:与官方版本同步,查询该软件的版本号,例如2.0.0;
## Note
+ 若使用 pip install 下载安装过慢,可添加源:-i https://pypi.tuna.tsinghua.edu.cn/simple/
......
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