diff --git a/ci/docker/toolchains/aarch64-linux-gnu-toolchain.cmake b/ci/docker/toolchains/aarch64-linux-gnu-toolchain.cmake index 3780415c4b15..57072b5cf393 100644 --- a/ci/docker/toolchains/aarch64-linux-gnu-toolchain.cmake +++ b/ci/docker/toolchains/aarch64-linux-gnu-toolchain.cmake @@ -19,6 +19,7 @@ set(CMAKE_SYSTEM_NAME Linux) set(CMAKE_SYSTEM_PROCESSOR "aarch64") set(CMAKE_C_COMPILER aarch64-linux-gnu-gcc) set(CMAKE_CXX_COMPILER aarch64-linux-gnu-g++) +set(CMAKE_CUDA_COMPILER nvcc) set(CMAKE_CUDA_HOST_COMPILER aarch64-linux-gnu-gcc) set(CMAKE_FIND_ROOT_PATH "/usr/aarch64-linux-gnu")