Commit 355115f0 authored by huchen's avatar huchen
Browse files

Update README.md

parent 02be4bc7
...@@ -15,9 +15,9 @@ pip install mmcv* (下载的mmcv的whl包) ...@@ -15,9 +15,9 @@ pip install mmcv* (下载的mmcv的whl包)
#### 编译环境准备 #### 编译环境准备
提供2种环境准备方式: 提供2种环境准备方式:
1基于光源pytorch基础镜像环境:镜像下载地址:[https://sourcefind.cn/#/image/dcu/pytorch](https://sourcefind.cn/#/image/dcu/pytorch)根据pytorch、python、dtk及系统下载对应的镜像版本。 1. 基于光源pytorch基础镜像环境:镜像下载地址:[https://sourcefind.cn/#/image/dcu/pytorch](https://sourcefind.cn/#/image/dcu/pytorch)根据pytorch、python、dtk及系统下载对应的镜像版本。
2基于现有python环境:安装pytorch,pytorch whl包下载目录:[https://cancon.hpccube.com:65024/4/main/pytorch/dtk23.04](https://cancon.hpccube.com:65024/4/main/pytorch/dtk23.04),根据python、dtk版本,下载对应pytorch的whl包。安装命令如下: 2. 基于现有python环境:安装pytorch,pytorch whl包下载目录:[https://cancon.hpccube.com:65024/4/main/pytorch/dtk23.04](https://cancon.hpccube.com:65024/4/main/pytorch/dtk23.04),根据python、dtk版本,下载对应pytorch的whl包。安装命令如下:
```shell ```shell
pip install torch* (下载的torch的whl包) pip install torch* (下载的torch的whl包)
pip install setuptools=59.5.0 wheel pip install setuptools=59.5.0 wheel
...@@ -30,11 +30,11 @@ git clone https://developer.hpccube.com/codes/aicomponent/mmcv # 根据编译需 ...@@ -30,11 +30,11 @@ git clone https://developer.hpccube.com/codes/aicomponent/mmcv # 根据编译需
``` ```
- 提供2种源码编译方式(进入mmcv目录): - 提供2种源码编译方式(进入mmcv目录):
``` ```
1 编译whl包并安装 1. 编译whl包并安装
MMCV_WITH_OPS=1 ROCM_HOME=${ROCM_PATH} python3 setup.py -v bdist_wheel MMCV_WITH_OPS=1 ROCM_HOME=${ROCM_PATH} python3 setup.py -v bdist_wheel
pip install dist/mmcv* pip install dist/mmcv*
2 源码编译安装 2. 源码编译安装
MMCV_WITH_OPS=1 ROCM_HOME=${ROCM_PATH} python3 setup.py install MMCV_WITH_OPS=1 ROCM_HOME=${ROCM_PATH} python3 setup.py install
``` ```
......
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