深度学习训练模型时,出现以下错误:
RuntimeError: CUDA error: CUBLAS_STATUS_NOT_SUPPORTED when calling `cublasSgemmStridedBatched( handle, opa, opb, m, n, k, &alpha, a, lda, stridea, b, ldb, strideb, &beta, c, ldc, stridec, num_batches)`
尝试输入下面命令:
export CUDA_HOME=/usr/local/cuda
或者
unset LD_LIBRARY_PATH