由於重重原因我把機子的環境中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