win10下tensorflow1.3升級到1.6
阿新 • • 發佈:2019-02-10
1. CUDA解除安裝,Windows開啟控制面板,除了NVIDIA影象驅動程式和NVIDIA系統兩個,其他有關NVIDIA的軟體全部解除安裝;
2、安裝CUDA9.0,一路預設,安裝過程中若出現:could not fine compatible graphic hardware
繼續安裝,直接點選繼續—>同意並繼續—>自定義(高階)—>只選擇CUDA進行安裝(即第一個選項),最後安裝成功。(cmd以後,用nvcc -V檢視,例如下圖內容,即成功),若提示沒有nvcc指令,將控制檯重新關閉,在開啟就可以了)
3、安裝CUDNN,這是NVIDIA推出的一個利用GPU加速卷積神經網路的庫。把解壓出來的檔案覆蓋到CUDA安裝路徑即可(預設:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0);
4、解除安裝原來的tensorflow:pip uninstall tensorflow-gpu
5、安裝新版本的tensorflow:pip install tensorflow-gpu==1.6
6、在TensorFlow目錄下,安裝Object Detection API:
git clone https://github.com/tensorflow/models.git
7、在tensorFlow/models/research目錄下執行protoc編譯命令:
protoc object_detection/protos/*.proto --python_out=.
8.、測試安裝是否OK:
1). 開啟命令列,輸入python,進入python互動模式 ;2). 輸入命令:import tensorflow; print(tensorflow.__version__)
3). 如果輸出:1.6.0則安裝成功,否則安裝失敗。