Commit 7fc800d9 authored by limm's avatar limm
Browse files

updated README.md support for fastpt compilation

parent b22ab379
...@@ -24,9 +24,9 @@ pip install mmcv* (下载的mmcv的whl包) ...@@ -24,9 +24,9 @@ pip install mmcv* (下载的mmcv的whl包)
#### 编译环境准备 #### 编译环境准备
提供基于fastpt不转码编译: 提供基于fastpt不转码编译:
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),根据pytorch、python、dtk及系统下载对应的镜像版本。
2. 基于现有python环境:安装pytorch,fastpt whl包下载目录:[http://10.6.10.68:8000/debug/pytorch/dtk24.04.1/](http://10.6.10.68:8000/debug/pytorch/dtk24.04.1/),根据python、dtk版本,下载对应pytorch的whl包。安装命令如下: 2. 基于现有python环境:安装pytorch,fastpt whl包下载目录:[http://10.16.4.1:8000/debug/pytorch/dtk25.04/](http://10.16.4.1:8000/debug/pytorch/dtk25.04/),根据python、dtk版本,下载对应pytorch的whl包。安装命令如下:
```shell ```shell
pip install torch* (下载的torch的whl包) pip install torch* (下载的torch的whl包)
pip install fastpt* (下载的fastpt的whl包, 安装顺序, 先安装torch,后安装fastpt) pip install fastpt* (下载的fastpt的whl包, 安装顺序, 先安装torch,后安装fastpt)
...@@ -41,8 +41,8 @@ git clone https://developer.hpccube.com/codes/aicomponent/mmcv # 根据编译需 ...@@ -41,8 +41,8 @@ git clone https://developer.hpccube.com/codes/aicomponent/mmcv # 根据编译需
- 提供2种源码编译方式(进入mmcv目录): - 提供2种源码编译方式(进入mmcv目录):
``` ```
1. 设置不转码编译环境变量 1. 设置不转码编译环境变量
source /opt/dtk/cuda/env.sh export FORCE_CUDA=1
export USE_FASTPT_CUDA=True source /usr/local/bin/fastpt -C
2. 编译whl包并安装 2. 编译whl包并安装
MMCV_WITH_OPS=1 python3 setup.py -v bdist_wheel MMCV_WITH_OPS=1 python3 setup.py -v bdist_wheel
......
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