提交 3af33c63 authored 作者: vesis84's avatar vesis84

Revert "force-disable of GPU initialization with pycuda"

This reverts commit 1814576c.
上级 1814576c
...@@ -399,9 +399,9 @@ def use(device, ...@@ -399,9 +399,9 @@ def use(device,
# Has PyCUDA already initialized the GPU context # Has PyCUDA already initialized the GPU context
pycuda_init_dev = False pycuda_init_dev = False
# if config.pycuda.init: if config.pycuda.init:
# import theano.misc.pycuda_init import theano.misc.pycuda_init
# pycuda_init_dev = theano.misc.pycuda_init.pycuda_available pycuda_init_dev = theano.misc.pycuda_init.pycuda_available
try: try:
if (device != 'gpu') and not pycuda_init_dev: if (device != 'gpu') and not pycuda_init_dev:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论