caffe: fix python build

2019-07-20 13:17:17 +02:00
parent 5f597d4b4c
commit 955ed925e5
2 changed files with 2 additions and 2 deletions
@@ -53,7 +53,7 @@ stdenv.mkDerivation rec {
  cmakeFlags =
    # It's important that caffe is passed the major and minor version only because that's what
    # boost_python expects
-    [ (if pythonSupport then "-Dpython_version=3${python.pythonVersion}" else "-DBUILD_python=OFF")
+    [ (if pythonSupport then "-Dpython_version=${python.pythonVersion}" else "-DBUILD_python=OFF")
      "-DBLAS=open"
    ] ++ (if cudaSupport then [
           "-DCUDA_ARCH_NAME=All"