Skip to content

Commit 4485314

Browse files
committed
Update README.md
1 parent 32c6039 commit 4485314

File tree

1 file changed

+33
-1
lines changed

1 file changed

+33
-1
lines changed

README.md

Lines changed: 33 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -557,6 +557,8 @@ You can build the CPU container with:
557557
docker build .
558558
```
559559

560+
### CUDA - Pre Blackwell architecture
561+
560562
To build the CUDA containers, you need to know the compute cap of the GPU you will be using
561563
at runtime.
562564

@@ -581,10 +583,40 @@ runtime_compute_cap=89
581583
# Example for H100
582584
runtime_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, ...)
585617
runtime_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

0 commit comments

Comments
 (0)