运行不成功

[245680]

似乎搜索正常,总结报error

RuntimeError: CUDA error: CUBLAS_STATUS_NOT_SUPPORTED when calling cublasLtMatmulAlgoGetHeuristic( ltHandle, computeDesc.descriptor(), Adesc.descriptor(), Bdesc.descriptor(), Cdesc.descriptor(), Cdesc.descriptor(), preference.descriptor(), 1, &heuristicResult, &returnedResult)

最好给个直接能用的docker镜像,配环境太麻烦了

[ilovesouthpark]

RuntimeError: CUDA error: CUBLAS STATUS NOT SUPPORTED when calling cublasGemmix( handle, opa,
opb, m,n,.k,&falpha
CUDA R 16F，1da,i b,CUDA R 16F, ldb, &fbeta, C, CUDA R 16F, 1dc, CUDA R 32F,CUBLAS GEHM DFALT-TENSOR OP)。也是类似错误，发生在最后进行总结的时候。是因为使用的cuda版本的问题还是什么问题？

[ilovesouthpark]

补充一下，4090单卡，torch 2.0.1，cuda 11.8，cudnn 8700，anacoda建立的虚拟环境

[flyforfreedom]

能有个docker镜像就方便多了。

[wjy3326]

同样出现这个问题了,请问是什么原因?问题解决了吗?