表題の件なのですが、UbuntuにDeepLearningの環境を構築しようと思い、CUDA、cudnnなどをインストールしていざmnistでgpuを動かそうとしたのですが以下のようなエラーでGPUが使えず困っています。
エラー内容からDropoutがダメなのかと思い、trainをFalseにすると動きましたがTrueにすると再び以下のエラーが出ます。
このようなエラーの対処法をご存知の方がいらっしゃれば教えていただきたいです。
自分の環境は
Ubuntu16.04
CUDA8.0
cudnn5.1
Chainer1.24.0
です。
エラーの内容と使用しているソースを載せますのでよろしくお願いいたします。
fetch MNIST dataset
('epoch', 1)
Traceback (most recent call last):
File "/home/hoge/ドキュメント/mnist.py", line 89, in <module>
loss, acc = model(x_batch, y_batch)
File "/home/hoge/ドキュメント/mnist.py", line 30, in __call__
h1 = F.dropout(F.maxout(self.l1(x), 1), train=train, ratio = 0.0)
File "/usr/local/lib/python2.7/dist-packages/chainer/functions/noise/dropout.py", line 55, in dropout
return Dropout(ratio)(x)
File "/usr/local/lib/python2.7/dist-packages/chainer/function.py", line 199, in __call__
outputs = self.forward(in_data)
File "/usr/local/lib/python2.7/dist-packages/chainer/functions/noise/dropout.py", line 26, in forward
flag = (xp.random.rand(*x[0].shape, dtype=numpy.float32) >=
File "/usr/local/lib/python2.7/dist-packages/cupy/random/sample.py", line 28, in rand
return random_sample(size=size, dtype=dtype)
File "/usr/local/lib/python2.7/dist-packages/cupy/random/sample.py", line 136, in random_sample
return rs.random_sample(size=size, dtype=dtype)
File "/usr/local/lib/python2.7/dist-packages/cupy/random/generator.py", line 152, in random_sample
func(self._generator, out.data.ptr, out.size)
File "cupy/cuda/curand.pyx", line 148, in cupy.cuda.curand.generateUniform (cupy/cuda/curand.cpp:2847)
File "cupy/cuda/curand.pyx", line 151, in cupy.cuda.curand.generateUniform (cupy/cuda/curand.cpp:2749)
File "cupy/cuda/curand.pyx", line 85, in cupy.cuda.curand.check_status (cupy/cuda/curand.cpp:1437)
cupy.cuda.curand.CURANDError: CURAND_STATUS_LAUNCH_FAILURE