Skip to content

Commit c406619

Browse files
committed
Update custom_container.md docs
1 parent 4485314 commit c406619

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

docs/source/en/custom_container.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ rendered properly in your Markdown viewer.
1616

1717
# Build a custom container for TEI
1818

19-
You can build our own CPU or CUDA TEI container using Docker. To build a CPU container, run the following command in the
19+
You can build our own CPU or CUDA TEI container using Docker. To build a CPU container, run the following command in the
2020
directory containing your custom Dockerfile:
2121

2222
```shell
@@ -35,7 +35,8 @@ the examples of runtime compute capabilities for various GPU types:
3535
- Blackwell (RTX 5000 series, ...) - `runtime_compute_cap=120`
3636

3737
Once you have determined the compute capability is determined, set it as the `runtime_compute_cap` variable and build
38-
the container as shown in the example below:
38+
the container using `Dockerfile-cuda` if the runtime compute cap is lower than 120 otherwise use
39+
`Dockerfile-cuda-blackwell` as shown in the example below:
3940

4041
```shell
4142
# Get submodule dependencies

0 commit comments

Comments
 (0)