最近在服務器上開始跑TensorFlow的實驗,遇到了一個問題,在這裏記錄一下解決過程。就是雖然安裝了TensorFlow-gpu,但是在import的時候卻會報錯,具體如下所示
經過查詢,看到GitHub上有人說是cuda版本的問題。傳送門:https://github.com/tensorflow/tensorflow/issues/15656
上面也有人說TensorFlow 1.5已經能夠在cuda 9.1上跑了,但那可能是Windows系統或者別的什麼原因,總之我這裏還是不行。所以只有重裝了cuda 9.0才解決了問題。
查看cuda版本的方法:https://blog.csdn.net/Teeyohuang/article/details/79082003
安裝cuda的方法:https://cloud.tencent.com/document/product/560/8064