Ubuntu18.04+CUDA9.1+cuDNN7.0.5下安裝tensorflow-gpu版本

由於我的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()也不報錯。

這就證明安裝好了。

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