Ubuntu16.04安裝NVIDIA顯卡驅動和cuda9、cudnn7過程
硬件及軟件搭配
系統:ubuntu16.04.5
NVIDIA顯卡驅動版本:384.130
cuda版本:9.0
cudnn:cudnn7.0.5
tensorflow-gpu:1.8
安裝過程:
系統:ubuntu16.04.5:
製作啓動盤,選擇分區,
NVIDIA顯卡驅動版本:384.130
更換阿里雲的apt源,使用可視化的方法:
參考:https://blog.csdn.net/qq_34782535/article/details/88032485#4aptget_27
選擇之後關閉,確認重新載入,更新軟件列表,後可以在附加驅動裏看到
使用英偉達384顯卡驅動,應用更改等待完成重啓
還有其他辦法:
(1)通過官方run文件安裝驅動,或同時安裝驅動和cuda;這種方法,需要警用nouveau;我一直禁用不成功,所以放棄
(2)通過ppa安裝,好像下載速度很慢,沒試過
安裝完成後:
cuda版本:9.0和cudnn版本
驅動版本、cuda版本、cudnn、tensorflow之間有版本對應關係,不可以隨便安裝:
cuda9.1就要求驅動版本大於等於387.26;所以這裏選擇cuda9
cuda9可支持cudnn7.6,但是保險起見使用整數版本cudnn7.0
具體可參考
https://blog.csdn.net/qq_30163461/article/details/80324843
其他遇到的坑
安裝python3-pip:
報依賴問題
gnupg : 破壞: software-properties-common (<= 0.96.24.3) 但 0.96.20.7要求被安裝
解決方法:換源的方法採用以上可視化方法
終端使用python3,導入tensorflow成功,但是pycharm不行:
之前沒遇到過,但是後來發現,在IDE添加一下cuda路徑即可