提交 d8a04356 authored 作者: notoraptor's avatar notoraptor

Update `theano-cache` to prevent an error message printed when theano

flag `device` is set to `cuda` on Windows.
上级 673a6713
......@@ -5,11 +5,11 @@ import os
import sys
if sys.platform == 'win32':
config_cxx = 'cxx='
config_for_theano_cache_script = 'cxx=,device=cpu'
theano_flags = os.environ['THEANO_FLAGS'] if 'THEANO_FLAGS' in os.environ else ''
if theano_flags:
theano_flags += ','
theano_flags += config_cxx
theano_flags += config_for_theano_cache_script
os.environ['THEANO_FLAGS'] = theano_flags
import theano
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论