Tensorflow-GPU[解決Tensorflow:ImportError:libcusolver.so.9.0]

由於重重原因我把機子的環境中CUDA9.0+CUDNN7.0.5+tensorflow-GPU1.6的變成了:CUDA8.0+cudnn6.0+tensorflow-gpu1.6
之後在引入了:

import tensorflow

的時候就拋出異常:

Tensorflow:ImportError:libcusolver.so.9.0

剛開始還很納悶,還以爲是CUDA沒有卸載乾淨,又重新卸載+安裝,可是還是這個問題。嘗試了tensorflow-gpu的1.5、1.6、1.7不行,最終安裝了tensorflow-gpu1.4版本就可以了。

結論就是:tensorflow-gpu1.5以後不支持使用CUDA8.0

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章