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
wxj
NeMo
Commits
c969d6d1
Commit
c969d6d1
authored
Jan 07, 2025
by
wxj
Browse files
Update README.md
parent
0651a856
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
43 additions
and
1 deletion
+43
-1
README.md
README.md
+43
-1
No files found.
README.md
View file @
c969d6d1
最新测试镜像: torch2.4.1-py3.10-dtk25.04-beta-das-alpha
该镜像自带transformer_engine1.8
git下载该项目
启动容器:
```
bash
docker run
-it
\
--shm-size
=
32G
\
--device
=
/dev/kfd
\
--device
=
/dev/mkfd
\
--device
=
/dev/dri
\
--cap-add
=
SYS_PTRACE
\
--security-opt
seccomp
=
unconfined
\
--ulimit
memlock
=
-1
:-1
\
--ipc
=
host
\
--network
=
host
\
--group-add
video
\
--privileged
\
--name
nemo_dtk25.4
\
-v
/opt/hyhal:/opt/hyhal
\
-v
/path/to/data/:/data
\
-v
/path/to/workspace/:/workspace
\
ce83b4a462d9
\
/bin/bash
```
安装依赖
```
bash
cd
nemo_dtk25-2.0.0.rc0.beta
pip
install
-r
requirements.txt
-i
https://pypi.tuna.tsinghua.edu.cn/simple
pip
install
.
-i
https://pypi.tuna.tsinghua.edu.cn/simple
cd
..
&&
cd
Megatron-LM-core_r0.7.0.beta
pip
install
.
-i
https://pypi.tuna.tsinghua.edu.cn/simple
```
执行微调脚本:
单机八卡:
`bash K100AI_finetune.sh >& K100AI_finetune.log`
readme
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