tensorflow不斷升級,最新版本的程式碼需要更高版本的cuda支援。以前一直用cuda8.0和cudnn6.0的組合,tensorflow用1.3.0版本。不過最新下載的models,某些程式碼執行時,已經出現了錯誤。
這裡寫圖片描述
類似的錯誤是因為tensorflow升級,models中呼叫了新版本的函式,而安裝的tensorflow是舊版本。為了執行程式碼,需要更新tensorflow至新版本。而tensorflow新版本,又要求更高版本的cuda和cudnn。因此這裡簡單說一下解除安裝nvidia驅動和cuda的過程,至於重新安裝,可以參考部落格https://blog.csdn.net/ezhchai/article/details/69663082
1.解除安裝nvidia驅動
解除安裝NV驅動和安裝一樣,首先ctrl+Alt+F2進入命令列狀態,然後停止lightdm
sudo service lightdm stop
或者
sudo /etc/init.d/lightdm stop

解除安裝命令位置/usr/bin/nvidia-uninstall,以下命令即可解除安裝。
sudo /usr/bin/nvidia-uninstall
不找這個命令的位置,也可以
sudo apt-get install autoremove --purge nvidia*
這裡寫圖片描述
這樣就完成了解除安裝。

2.解除安裝cuda
cuda的預設安裝在 /usr/local/cuda-8.0下,用下面的命令解除安裝:
sudo /usr/local/cuda-8.0/bin/uninstall_cuda-8.0.pl
會出現長時間的刷屏現象,說明cuda檔案其實還蠻多的~~
這裡寫圖片描述
/usr/local/cuda-8.0沒有自動刪除,需要手動完成刪除。其實刪不刪無所謂,知道它沒用了也就是了。
這裡寫圖片描述
解除安裝之後,重啟一下。可以看到解析度有所不同,這是按照https://blog.csdn.net/ezhchai/article/details/69663082中描述的方法,重新安裝cuda9.0和cudnn7.0即可。