在Windows conda上使用GPU设备导入theano

时间:2018-03-21 00:58:33

标签: python anaconda theano

我在使用带Conda for Windows的工作站时遇到了一些麻烦。我对操作系统不太熟悉,这是我第一次尝试GPU支持那里的asano,但没有用。

问题是,当我使用Anaconda bash时,我可以说:

set "MKL_THREADING_LAYER=GNU" set THEANO_FLAGS=device=cuda python import theano

在GPU支持下,这很好用。但是,我需要脚本在执行期间在设备(GPU和CPU)之间切换。我在某处读到了可以通过直接在代码上设置环境变量来完成,但我试过这个无济于事:

import os os.environ["THEANO_FLAGS"] = "device=cuda" import theano

MKL_THREADING_LAYER环境变量已经放入系统中,所以我猜错误不在那里。无论如何,代码无法运行:

RuntimeError: 'path' must be None or a list, not <class '_frozen_importlib_external._NamespacePath'>

有什么想法吗?感谢。

0 个答案:

没有答案