3
0
Fork 0
forked from mirrors/nixpkgs

Merge pull request #8644 from jb55/caffe-patch

caffe: init at git-rev 77d66dfc
This commit is contained in:
Arseniy Seroka 2015-07-08 20:35:41 +03:00
commit f56344eb74
2 changed files with 74 additions and 0 deletions

View file

@ -0,0 +1,70 @@
{ stdenv
, openblas
, boost
, cudaSupport ? true
, cudnnSupport ? false
, cudnn ? null
, cudatoolkit7
, fetchFromGitHub
, google-gflags
, glog
, hdf5
, leveldb
, lmdb
, opencv
, protobuf
, snappy
}:
let optional = stdenv.lib.optional;
in stdenv.mkDerivation rec {
# Use git revision because latest "release" is really old
name = "caffe-git-2015-07-02";
src = fetchFromGitHub {
owner = "BVLC";
repo = "caffe";
rev = "77d66dfc907dd875d69bb9fc12dd950b531e464f";
sha256 = "0vd4qrc49dhsawj298xpkd5mvi35sh56kdswx3yp8ya4fjajwakx";
};
preConfigure = "mv Makefile.config.example Makefile.config";
makeFlags = "BLAS=open " +
(if !cudaSupport then "CPU_ONLY=1 " else "CUDA_DIR=${cudatoolkit7} ") +
(if cudnnSupport then "USE_CUDNN=1 " else "");
# too many issues with tests to run them for now
doCheck = false;
checkPhase = "make runtest ${makeFlags}";
buildInputs = [ openblas boost google-gflags glog hdf5 leveldb lmdb opencv
protobuf snappy ]
++ optional cudaSupport cudatoolkit7
++ optional cudnnSupport cudnn;
installPhase = ''
mkdir -p $out/{bin,share,lib}
for bin in $(find build/tools -executable -type f -name '*.bin');
do
cp $bin $out/bin/$(basename $bin .bin)
done
cp -r build/examples $out/share
cp -r build/lib $out
'';
meta = with stdenv.lib; {
description = "Deep learning framework";
longDescription = ''
Caffe is a deep learning framework made with expression, speed, and
modularity in mind. It is developed by the Berkeley Vision and Learning
Center (BVLC) and by community contributors.
'';
homepage = http://caffe.berkeleyvision.org/;
maintainers = with maintainers; [ jb55 ];
license = licenses.bsd2;
platforms = platforms.linux;
};
}

View file

@ -14401,6 +14401,10 @@ let
### SCIENCE / MATH
caffe = callPackage ../applications/science/math/caffe {
cudaSupport = config.caffe.cudaSupport or config.cudaSupport or true;
};
ecm = callPackage ../applications/science/math/ecm { };
eukleides = callPackage ../applications/science/math/eukleides {