CUBLAS_STATUS_INVALID_VALUE when calling `cublasGemmStridedBatchedExFix( handle, opa, opb, m, n, k, (void*)(&falpha), a, CUDA_R_16F, lda, stridea, b, CUDA_R_16F, ldb, strideb, (void*)(&fbeta), c, CUDA_R_16F, ldc, stridec,
之前跑通的代码突然不能用了,在网上找了一圈也没找到答案
最后发现是安装其他库的时候自动装了一个cuda 包导致的

直接
pip uninstall nvidia-cublas-cu11
程序成功运行
博客讲述了原本能跑通的代码突然无法运行,在网上查找无果后,发现是安装其他库时自动安装的cuda包导致的问题,解决该问题后程序成功运行,涉及深度学习相关内容。
8万+

被折叠的 条评论
为什么被折叠?



