提交 214128e0 authored 作者: James Bergstra's avatar James Bergstra

updated theano-compiledir to use modulecache

上级 990f46fd
#!/usr/bin/env python #!/usr/bin/env python
from theano.gof import cc
import sys import sys
compiledir = cc.get_compiledir() from theano.gof.cc import get_compiledir, get_module_cache
if len(sys.argv) == 1: if len(sys.argv) == 1:
print compiledir print get_compiledir()
elif sys.argv[1] in ('clear'): elif sys.argv[1] in ('clear'):
cc.clear_compiledir() get_module_cache().clear()
else: else:
print 'command "%s" not recognized' % sys.argv[1] print 'command "%s" not recognized' % sys.argv[1]
print 'Type "theano-cache" to print the cache location'
print 'Type "theano-cache clear" to erase the cache'
sys.exit(1) sys.exit(1)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论