cuda 10.2のサポートについて

[Yokokawa Hideaki]

お世話になっております

当方、Quadro P6000を使用しております。

また、OSはubuntu18.04LTSです。

apt-get upgradeを行ったところ、どうやらcuda10.2が入ったようで、

cupyがうまく動かないようです。

以下、エラーの吐き出しを貼ります。

--

Traceback (most recent call last):

    y,h=model(bat_data)

  File "/usr/local/lib/python3.6/dist-packages/chainer/link.py", line 294, in __call__

    out = forward(*args, **kwargs)

  File "TEST.py", line 36, in forward

    h = F.tanh(self.l1(x))

  File "/usr/local/lib/python3.6/dist-packages/chainer/link.py", line 294, in __call__

    out = forward(*args, **kwargs)

  File "/usr/local/lib/python3.6/dist-packages/chainer/links/connection/linear.py", line 155, in forward

    return linear.linear(x, self.W, self.b, n_batch_axes=n_batch_axes)

  File "/usr/local/lib/python3.6/dist-packages/chainer/functions/connection/linear.py", line 308, in linear

    y, = LinearFunction().apply(args)

  File "/usr/local/lib/python3.6/dist-packages/chainer/function_node.py", line 321, in apply

    outputs = self.forward(in_data)

  File "/usr/local/lib/python3.6/dist-packages/chainer/functions/connection/linear.py", line 116, in forward

    outputs=[y])

  File "/usr/local/lib/python3.6/dist-packages/chainer/graph_optimizations/static_graph_utilities.py", line 184, in wrapped_func

    ret = func(*args, **kwargs)

  File "/usr/local/lib/python3.6/dist-packages/chainer/functions/connection/linear.py", line 56, in static_linear_no_bias

    xp.dot(x, W.T, out=y)

  File "/usr/local/lib/python3.6/dist-packages/cupy/linalg/product.py", line 35, in dot

    return a.dot(b, out)

  File "cupy/core/core.pyx", line 1326, in cupy.core.core.ndarray.dot

  File "cupy/core/core.pyx", line 1988, in cupy.core.core.dot

  File "cupy/core/core.pyx", line 2357, in cupy.core.core.tensordot_core

  File "cupy/cuda/device.pyx", line 46, in cupy.cuda.device.get_cublas_handle

  File "cupy/cuda/device.pyx", line 47, in cupy.cuda.device.get_cublas_handle

  File "cupy/cuda/device.pyx", line 187, in cupy.cuda.device.Device.cublas_handle.__get__

  File "cupy/cuda/device.pyx", line 174, in cupy.cuda.device.Device._get_handle

  File "cupy/cuda/device.pyx", line 175, in cupy.cuda.device.Device._get_handle

  File "cupy/cuda/cublas.pyx", line 315, in cupy.cuda.cublas.create

  File "cupy/cuda/cublas.pyx", line 319, in cupy.cuda.cublas.create

  File "cupy/cuda/cublas.pyx", line 308, in cupy.cuda.cublas.check_status

cupy.cuda.cublas.CUBLASError: CUBLAS_STATUS_NOT_INITIALIZED

エラーの様子をみると、cupyとcudaがうまくつながっていないように見えます。

対応の方法があれば、ご教授お願いいたします。

次期アップデートで対応をお願いいたしたく思います

宜しくお願いいたします。