Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
MMCV
Commits
7fc800d9
Commit
7fc800d9
authored
May 28, 2025
by
limm
Browse files
updated README.md support for fastpt compilation
parent
b22ab379
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
README.md
README.md
+4
-4
No files found.
README.md
View file @
7fc800d9
...
...
@@ -24,9 +24,9 @@ pip install mmcv* (下载的mmcv的whl包)
#### 编译环境准备
提供基于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/dtk2
4
.04
.1
/
](
http://10.6.
10.68
:8000/debug/pytorch/dtk2
4
.04
.1
/
)
,根据python、dtk版本,下载对应pytorch的whl包。安装命令如下:
2.
基于现有python环境:安装pytorch,fastpt whl包下载目录:
[
http://10.
1
6.
4.1
:8000/debug/pytorch/dtk2
5
.04/
](
http://10.
1
6.
4.1
:8000/debug/pytorch/dtk2
5
.04/
)
,根据python、dtk版本,下载对应pytorch的whl包。安装命令如下:
```
shell
pip
install
torch
*
(
下载的torch的whl包
)
pip
install
fastpt
*
(
下载的fastpt的whl包, 安装顺序, 先安装torch,后安装fastpt
)
...
...
@@ -41,8 +41,8 @@ git clone https://developer.hpccube.com/codes/aicomponent/mmcv # 根据编译需
-
提供2种源码编译方式(进入mmcv目录):
```
1. 设置不转码编译环境变量
source /opt/dtk/cuda/env.sh
export USE_FASTPT_CUDA=True
export FORCE_CUDA=1
source /usr/local/bin/fastpt -C
2. 编译whl包并安装
MMCV_WITH_OPS=1 python3 setup.py -v bdist_wheel
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment