diff options
Diffstat (limited to 'pkgs/development/libraries/science/math/caffe2/default.nix')
-rw-r--r-- | pkgs/development/libraries/science/math/caffe2/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/libraries/science/math/caffe2/default.nix b/pkgs/development/libraries/science/math/caffe2/default.nix index d568e945202..4746f77a216 100644 --- a/pkgs/development/libraries/science/math/caffe2/default.nix +++ b/pkgs/development/libraries/science/math/caffe2/default.nix @@ -1,6 +1,6 @@ { stdenv, lib, config, fetchFromGitHub , cmake -, glog, google-gflags, gtest +, glog, gflags, gtest , protobuf, snappy , python, future, six, python-protobuf, numpy, pydot , eigen @@ -74,7 +74,7 @@ stdenv.mkDerivation rec { outputs = [ "bin" "out" ]; propagatedBuildOutputs = [ ]; # otherwise propagates out -> bin cycle - buildInputs = [ glog google-gflags protobuf snappy eigen ] + buildInputs = [ glog gflags protobuf snappy eigen ] ++ lib.optional useCuda cudatoolkit ++ lib.optional useCudnn cudnn ++ lib.optional useOpenmp openmp |