TensorFlow: Blas GEMM launch failed

后端 未结 15 1121
攒了一身酷
攒了一身酷 2020-12-05 05:11

When I\'m trying to use TensorFlow with Keras using the gpu, I\'m getting this error message:

C:\\Users\\nicol\\Anaconda3\\envs\\tensorflow\\lib\\site-packag         


        
15条回答
  •  春和景丽
    2020-12-05 05:22

    Try running the sample program simpleCUBLAS (it comes with CUDA) to test your CUBLAS installation and see if it works.

    In my case (I am using Ubuntu) I had to reinstall CUDA to solve this issue. After I did that, simpleCUBLAS passed the test.

    For some reason I started running into the same issue after a while, and I found that cleaning the directory .nv (inside my home folder) resolved the issue, and simpleCUBLAS test passed again.

提交回复
热议问题