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