Skip to content

Commit ea69046

Browse files
authored
Add libcurand-dev to CUDA Docker image for cuda-windows CI (#19183)
Commit c1731fd added CUDA::curand as a CMake dependency in backends/cuda/CMakeLists.txt for the new GPU-side sampling shim (rand.cu). However, the CI Docker image (install_cuda.sh) did not include libcurand-dev, so the CMake build during pip install fails when EXECUTORCH_BUILD_CUDA=ON is auto-detected. This causes executorch to not be installed at all, breaking the dinov2 export (No module named 'executorch') and parakeet export (No module named 'torchaudio', since torchaudio installation is skipped when the prior pip install step fails) in the cuda-windows CI workflow.
1 parent 9207001 commit ea69046

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

.ci/docker/common/install_cuda.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,13 +38,15 @@ apt-get update
3838
# - libcublas-dev: cuBLAS development files
3939
# - libcusparse-dev: cuSPARSE development files
4040
# - libcufft-dev: cuFFT development files
41+
# - libcurand-dev: cuRAND development files
4142
apt-get install -y --no-install-recommends \
4243
"cuda-nvcc-${CUDA_VERSION_DASH}" \
4344
"cuda-cudart-dev-${CUDA_VERSION_DASH}" \
4445
"cuda-nvrtc-dev-${CUDA_VERSION_DASH}" \
4546
"libcublas-dev-${CUDA_VERSION_DASH}" \
4647
"libcusparse-dev-${CUDA_VERSION_DASH}" \
47-
"libcufft-dev-${CUDA_VERSION_DASH}"
48+
"libcufft-dev-${CUDA_VERSION_DASH}" \
49+
"libcurand-dev-${CUDA_VERSION_DASH}"
4850

4951
# Clean up
5052
apt-get clean

0 commit comments

Comments
 (0)