colmap: fix CUDA support

This commit is contained in:
Gaetan Lepage
2026-06-14 19:02:38 +00:00
parent 3dbdf4f055
commit bcb2907c62
+4 -5
View File
@@ -62,14 +62,12 @@ let
onnxruntime
]
++ lib.optionals cudaSupport [
cudatoolkit
(lib.getOutput "static" cudaPackages.cuda_cudart)
cudaPackages.cuda_cudart # CUDA::cudart, used by COLMAP and faiss
cudaPackages.libcublas # CUDA::cublas, propagated by faiss' exported targets
cudaPackages.libcurand # CUDA::curand, used by COLMAP
]
++ lib.optional stdenv'.cc.isClang llvmPackages.openmp;
# TODO: migrate to redist packages
inherit (cudaPackages) cudatoolkit;
# COLMAP needs these model files to run the ONNX tests
# Based on: https://github.com/colmap/colmap/blob/79efc74b2b614935a3c69b1f983f2bad23a836a1/src/colmap/feature/resources.h#L36
modelsForTesting = [
@@ -179,6 +177,7 @@ stdenv'.mkDerivation (finalAttrs: {
]
++ lib.optionals cudaSupport [
autoAddDriverRunpath
cudaPackages.cuda_nvcc
];
doCheck = enableTests;