由於我的NVIDIA 驅動是390:
所以我開始以爲安裝CUDA9.0(一直包CUDA和驅動版本不匹配),其實根據下表應該安裝CUDA9.1。
最後我安裝好了CUDA9.1+cuDNN7.0.5後安裝tensorflow-gpu死活裝不上,我試過tensorflow-gpu的版本有:1.5.0, 1.6.0 ,
1.7.0, 1.8.0, 1.10.0, 1.12.0, ...最後用1.6.0終於成功,import tensorflow as tf不報錯, 運行sess = tf.Session()也不報錯。
這就證明安裝好了。