BUG: 30系显卡tensorflow1.15 failed to run cuBLAS routine cublasSgemm_v2: CUBLAS_STATUS_EXECUTION_FAILED
系统情况
linux
3090ti 显卡
系统默认装了CUDA 11.4
原因与解决
因为我以前使用2080ti显卡,按照tensorflow官网要求1.15版本需要CUDA 10.0,然后我单独在当前python环境使用conda 安装了cuda 10.0 版本, 在2080ti这样做是没有问题的。
但我不知道英伟达30系已经不支持 10.0 版本的CUDA。导致模型代码运行失败弹出这个异常。
经过测试,tensorflow1.15 可以直接在CUDA 11.4版本运行的,我是直接通过下面命令安装即可在GPU(CUDA 11.4)运行。
安装命令
# gpu版
pip install tensorflow-gpu==1.15
所以我单独安装CUDA10.0是白费功夫了。。。
注:如果大家遇到在CUDA 11.4不能运行,我也不知道怎么办,请参考其他人的文章。

作者在尝试在3090ti显卡上运行tensorflow1.15时遇到错误,由于之前为2080ti显卡安装了CUDA10.0,而在30系显卡上CUDA10.0不兼容导致问题。实际上,tensorflow1.15可以与CUDA11.4配合使用,单独安装CUDA10.0是不必要的。若在CUDA11.4上遇到问题,作者建议参考其他资源解决。
1693

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



