tensorflow running error with cublas

后端 未结 6 939
無奈伤痛
無奈伤痛 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:58

    The compatibility issue between CUDA version and TensorFlow version. In my case, My CUDA version is 10.0 and TensorFlow version is 2.1.0, and this issue occurs. After changing TensorFlow 2.1.0 to TensorFlow 2.0.0, this issue disappears.

提交回复
热议问题