Skip to content

Commit aa1ee31

Browse files
author
JacoCheung
committed
Move root ReadMe env setting up to training
1 parent 7f8c695 commit aa1ee31

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

examples/hstu/training/README.md

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,14 @@ 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
89
docker build -f docker/Dockerfile --platform linux/amd64 -t recsys-examples:latest .
910
```
1011
If you want to build image for Grace, you can use
1112
```
13+
git clone https://github.com/NVIDIA/recsys-examples.git && cd recsys-examples
1214
docker build -f docker/Dockerfile --platform linux/arm64 -t recsys-examples:latest .
1315
```
1416
You can also set your own base image with args `--build-arg <BASE_IMAGE>`.

0 commit comments

Comments
 (0)