File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -3,20 +3,22 @@ We have supported both retrieval and ranking model whose backbones are HSTU laye
33## Environment Setup
44### Start from dockerfile
55
6- We provide [ dockerfile] ( ./docker/Dockerfile ) for users to build environment.
6+ We provide [ dockerfile] ( ../../.. /docker/Dockerfile ) for users to build environment.
77```
8+ git clone https://github.com/NVIDIA/recsys-examples.git && cd recsys-examples
89docker build -f docker/Dockerfile --platform linux/amd64 -t recsys-examples:latest .
910```
1011If you want to build image for Grace, you can use
1112```
13+ git clone https://github.com/NVIDIA/recsys-examples.git && cd recsys-examples
1214docker build -f docker/Dockerfile --platform linux/arm64 -t recsys-examples:latest .
1315```
1416You can also set your own base image with args ` --build-arg <BASE_IMAGE> ` .
1517
1618### Start from source file
1719Before running examples, build and install libs under corelib following instruction in documentation:
18- - [ HSTU attention documentation] ( ./corelib/hstu/README.md )
19- - [ Dynamic Embeddings documentation] ( ./corelib/dynamicemb/README.md )
20+ - [ HSTU attention documentation] ( .../../.. /corelib/hstu/README.md )
21+ - [ Dynamic Embeddings documentation] ( .../../.. /corelib/dynamicemb/README.md )
2022
2123On top of those two core libs, Megatron-Core along with other libs are required. You can install them via pypi package:
2224
You can’t perform that action at this time.
0 commit comments