My network: (cuda = 9, pytorch = 1.1.0)
input tensor -> GRU(it works) -> FullConnect(that was when pytorch crashed) --x--> output

Traceback (most recent call last):
  File "main.py", line 42, in <module>
    mod_train.train(network, device, dataloader, optimizer, loss_fn, i)
  File "/root/code/GPU/mod_train.py", line 20, in train
    outputs = model(inputs)
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/root/code/GPU/mod_network.py", line 47, in forward
    input = self.fc_0(input[:, -1, :].cuda())
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/linear.py", line 92, in forward
    return F.linear(input, self.weight, self.bias)
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/functional.py", line 1406, in linear
    ret = torch.addmm(bias, input, weight.t())
RuntimeError: cublas runtime error : the GPU program failed to execute at /opt/conda/conda-bld/pytorch_1556653215914/work/aten/src/THC/THCBlas.cu:259


(Is there any doc explaining special requirements for GPU training?)
How to solve this problem ?