90 lines
1.9 KiB
Nix
90 lines
1.9 KiB
Nix
{ stdenv
|
|
, lib
|
|
, fetchFromGitHub
|
|
, cmake
|
|
, pkg-config
|
|
, libpng
|
|
, libjpeg
|
|
, libwebp
|
|
, blas
|
|
, lapack
|
|
, config
|
|
, guiSupport ? false
|
|
, libX11
|
|
, sse4Support ? stdenv.hostPlatform.sse4_1Support
|
|
, avxSupport ? stdenv.hostPlatform.avxSupport
|
|
, cudaSupport ? config.cudaSupport
|
|
, cudaPackages
|
|
}@inputs:
|
|
(if cudaSupport then cudaPackages.backendStdenv else inputs.stdenv).mkDerivation rec {
|
|
pname = "dlib";
|
|
version = "19.24.2";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "davisking";
|
|
repo = "dlib";
|
|
rev = "v${version}";
|
|
sha256 = "sha256-Z1fScuaIHjj2L1uqLIvsZ7ARKNjM+iaA8SAtWUTPFZk=";
|
|
};
|
|
|
|
postPatch = ''
|
|
rm -rf dlib/external
|
|
'';
|
|
|
|
cmakeFlags = [
|
|
(lib.cmakeBool "USE_SSE4_INSTRUCTIONS" sse4Support)
|
|
(lib.cmakeBool "USE_AVX_INSTRUCTIONS" avxSupport)
|
|
(lib.cmakeBool "DLIB_USE_CUDA" cudaSupport)
|
|
] ++ lib.optionals cudaSupport [
|
|
(lib.cmakeFeature "DLIB_USE_CUDA_COMPUTE_CAPABILITIES" (builtins.concatStringsSep "," (with cudaPackages.flags; map dropDot cudaCapabilities)))
|
|
];
|
|
|
|
nativeBuildInputs = [
|
|
cmake
|
|
pkg-config
|
|
] ++ lib.optionals cudaSupport (with cudaPackages; [
|
|
cuda_nvcc
|
|
]);
|
|
|
|
buildInputs = [
|
|
libpng
|
|
libjpeg
|
|
libwebp
|
|
blas
|
|
lapack
|
|
]
|
|
++ lib.optionals guiSupport [ libX11 ]
|
|
++ lib.optionals cudaSupport (with cudaPackages; [
|
|
cuda_cudart.dev
|
|
cuda_cudart.lib
|
|
cuda_cudart.static
|
|
cuda_nvcc.dev
|
|
libcublas.dev
|
|
libcublas.lib
|
|
libcublas.static
|
|
libcurand.dev
|
|
libcurand.lib
|
|
libcurand.static
|
|
libcusolver.dev
|
|
libcusolver.lib
|
|
libcusolver.static
|
|
cudnn.dev
|
|
cudnn.lib
|
|
cudnn.static
|
|
cuda_cccl.dev
|
|
]);
|
|
|
|
passthru = {
|
|
inherit
|
|
cudaSupport cudaPackages
|
|
sse4Support avxSupport;
|
|
};
|
|
|
|
meta = with lib; {
|
|
description = "A general purpose cross-platform C++ machine learning library";
|
|
homepage = "http://www.dlib.net";
|
|
license = licenses.boost;
|
|
maintainers = with maintainers; [ christopherpoole ];
|
|
platforms = platforms.unix;
|
|
};
|
|
}
|