"vscode:/vscode.git/clone" did not exist on "e3b8a72291af1fc807ef99553b2b7fac20c6edf8"
README.md 1.14 KB
Newer Older
xiabo's avatar
xiabo committed
1
2
3
4
5
6
7
8
9
10
11
12
13
# MMCV

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

### 使用pip安装
mmcv 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包)
14
```
xiabo's avatar
xiabo committed
15
### 使用源码安装
16

xiabo's avatar
xiabo committed
17
18
19
20
21
#### 编译环境准备
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包)
22
23
```

xiabo's avatar
xiabo committed
24
25
```shell
pip install setuptools=59.5.0 wheel
Zaida Zhou's avatar
Zaida Zhou committed
26
27
```

xiabo's avatar
xiabo committed
28
29
30
31
32
33
#### 编译安装
```shell
git clone -b dtk-23.04_v1.6.1 https://developer.hpccube.com/codes/aicomponent/mmcv
cd mmcv
MMCV_WITH_OPS=1 ROCM_HOME=${ROCM_PATH} python3 setup.py -v bdist_wheel
pip install dist/mmcv*
Zaida Zhou's avatar
Zaida Zhou committed
34
35
```

xiabo's avatar
xiabo committed
36
37
38
## Note
+ 若使用 pip install 下载安装过慢,可添加源:-i https://pypi.tuna.tsinghua.edu.cn/simple/
+ ROCM_PATH为dtk的路径,默认为/opt/dtkxxx
39

xiabo's avatar
xiabo committed
40
41
42
## 其他参考
- [README_ORIGIN](README_ORIGIN.md)
- [README_zh-CN](README_zh-CN.md)