File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff 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
2020directory 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
3737Once 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
You can’t perform that action at this time.
0 commit comments