CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm

[lange-bleu]

I wanna get a 601 dim dvector,and follow the step "Train from scratch"

1. python preprocess.py ../LibriSpeech/train-clean-360 -o preprocessed
2. python train.py preprocessed train601

then there is an error Traceback to FILE "modules/dvector.py"
"""Forward a batch through network."""
lstm_outs, _ = self.lstm(inputs) # (batch, seg_len, dim_cell)
embeds = torch.tanh(self.embedding(lstm_outs)) # (batch, seg_len, dim_emb)
~~~~~~~~~~~~~~ <--- HERE

RuntimeError: CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling cublasSgemm( handle, opa, opb, m, n, k, &alpha, a, lda, b, ldb, &beta, c, ldc)

[lange-bleu]

env error with cuda