File tree Expand file tree Collapse file tree 1 file changed +33
-1
lines changed Expand file tree Collapse file tree 1 file changed +33
-1
lines changed Original file line number Diff line number Diff line change @@ -557,6 +557,8 @@ You can build the CPU container with:
557557docker build .
558558```
559559
560+ ### CUDA - Pre Blackwell architecture
561+
560562To build the CUDA containers, you need to know the compute cap of the GPU you will be using
561563at runtime.
562564
@@ -581,10 +583,40 @@ runtime_compute_cap=89
581583# Example for H100
582584runtime_compute_cap=90
583585
586+ docker build . -f Dockerfile-cuda --build-arg CUDA_COMPUTE_CAP=$runtime_compute_cap
587+ ```
588+
589+ ### CUDA - Blackwell architecture
590+
591+ To build the CUDA containers for the Blackwell architecture CUDA 12.9 is required, you need to use a different Dockerfile
592+ and set the compute cap to 120.
593+ This Dockerfile can still be used to build for previous architectures.
594+
595+ Commands to build the container:
596+
597+ ``` shell
598+ # Get submodule dependencies
599+ git submodule update --init
600+
601+ # Example for Turing (T4, RTX 2000 series, ...)
602+ runtime_compute_cap=75
603+
604+ # Example for A100
605+ runtime_compute_cap=80
606+
607+ # Example for A10
608+ runtime_compute_cap=86
609+
610+ # Example for Ada Lovelace (RTX 4000 series, ...)
611+ runtime_compute_cap=89
612+
613+ # Example for H100
614+ runtime_compute_cap=90
615+
584616# Example for Blackwell (RTX 5000 series, ...)
585617runtime_compute_cap=120
586618
587- docker build . -f Dockerfile-cuda --build-arg CUDA_COMPUTE_CAP=$runtime_compute_cap
619+ docker build . -f Dockerfile-cuda-blackwell --build-arg CUDA_COMPUTE_CAP=$runtime_compute_cap
588620```
589621
590622### Apple M1/M2 arm64 architectures
You can’t perform that action at this time.
0 commit comments