diff --git a/docs/build.md b/docs/build.md index 1ba582d9f..482dd515e 100644 --- a/docs/build.md +++ b/docs/build.md @@ -44,6 +44,18 @@ cmake .. -DSD_CUDA=ON cmake --build . --config Release ``` +### CUDA Architectures + +Different NVIDIA GPUs have different CUDA architecture: [CUDA GPU Compute Capability](https://developer.nvidia.com/cuda-gpus) + +To build for multiple GPU architectures (matching the GitHub CI releases), use `-DCMAKE_CUDA_ARCHITECTURES=90;89;86;80;75`: + +```shell +mkdir build && cd build +cmake .. -DSD_CUDA=ON -DCMAKE_CUDA_ARCHITECTURES=90;89;86;80;75 +cmake --build . --config Release +``` + ## Build with HipBLAS This provides GPU acceleration using AMD GPU. Make sure to have the ROCm toolkit installed.