diff options
-rw-r--r-- | var/spack/repos/builtin/packages/caffe/package.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/caffe/package.py b/var/spack/repos/builtin/packages/caffe/package.py index bf34ffcf83..a47e0a04ab 100644 --- a/var/spack/repos/builtin/packages/caffe/package.py +++ b/var/spack/repos/builtin/packages/caffe/package.py @@ -84,4 +84,8 @@ class Caffe(CMakePackage): '-DUSE_LEVELDB=%s' % ('+leveldb' in spec), '-DUSE_LMDB=%s' % ('+lmdb' in spec)] + if spec.satisfies('+python'): + version = spec['python'].version.up_to(1) + args.append('-Dpython_version=%s' % version) + return args |