TensorFlow: Blas GEMM launch failed

后端 未结 15 1137
攒了一身酷
攒了一身酷 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:31

    I was getting exactly the same error message. I realized that there was an error with my CUDA installation, specifically with the cuBLAS library.

    You can check if yours has the same problem by running the sample program simpleCUBLAS (it comes with the CUDA installation, you will probably find it in the CUDA home folder:$CUDA_HOME\samples\7_CUDALibraries\simpleCUBLAS)

    Try running this program. If the test fails, you have a problem with your CUDA installation. You should try to reinstall it. That's how I solved the same problem here.

    Renaming cublas64_10.dll to cublas64_100.dll may be a solution.

提交回复
热议问题