fix build command for caffe with openblas (#4299)
This commit is contained in:
parent
43edcfbf13
commit
785168f5a1
@ -73,7 +73,8 @@ class Caffe(CMakePackage):
|
||||
|
||||
def cmake_args(self):
|
||||
spec = self.spec
|
||||
args = ['-DBLAS={0}'.format(spec['blas'].name),
|
||||
args = ['-DBLAS={0}'.format('open' if spec['blas'].name == 'openblas'
|
||||
else spec['blas'].name),
|
||||
'-DCPU_ONLY=%s' % ('~gpu' in spec),
|
||||
'-DUSE_CUDNN=%s' % ('+gpu' in spec),
|
||||
'-DBUILD_python=%s' % ('+python' in spec),
|
||||
|
Loading…
Reference in New Issue
Block a user