提交 3cd079f3 authored 作者: Frederic's avatar Frederic

Fix the init of cublas handle when we don't specify the device.

上级 3a343790
......@@ -365,6 +365,8 @@ def use(device,
# event if another device is selected later.
cuda_ndarray.cuda_ndarray.CudaNdarray.zeros((2, 3))
use.device_number = active_device_number()
# This is needed to initialize the cublas handle.
gpu_init(use.device_number)
if test_driver:
import theano.sandbox.cuda.tests.test_driver
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论