1. 程式人生 > >win10下tensorflow1.3升級到1.6

win10下tensorflow1.3升級到1.6

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則安裝成功,否則安裝失敗。