tensorflow running error with cublas

后端 未结 6 958
無奈伤痛
無奈伤痛 2020-12-16 04:24

when I successfully install tensorflow on cluster, I immediately running mnist demo to check if it\'s going well, but here I came up with a problem. I don\'t know what is th

6条回答
  •  没有蜡笔的小新
    2020-12-16 04:52

    Aside from the mentioned solutions, this error also gets thrown when the CUBLAS version isn't compatible with the CUDA version. In my case, libclubas10 version 10.2.2.89-1 was incompatible with CUDA 10.1, so I had to downgrade:

    sudo apt-get install libcublas10=10.2.1.243-1 libcublas-dev=10.2.1.243-1 cuda-libraries-10-1 cuda-libraries-dev-10-1
    

提交回复
热议问题