為了讓CaffeTensorflowwindows中共存,所以CUDA要進行版本切換,因為Caffe需要使用CUDA 8.0Tensorflow-gpu使用CUDA9.0,但安裝了CUDA 8.0CUDA 9.0之後,Caffe在執行build時,CaffeCmakefile會自動偵測到CUDA9.0版本而無法運作,因為系CUDA 9.0設為主要GPU路徑,所以以下要將GPU主要偵測改回CUDA8.0


先開啟命令提示字元(cmd),並輸入 : nvcc --version,顯示當下的CUDA版本為9.0版。

需要藉由更改環境變數來切換偵測到之CUDA版本

開啟環境變數 : 本機(點右鍵內容) -> 進階系統設定 -> 環境變數

將紅色框框裡的CUDA_PATH更改成要偵測之版本。

(此處我從v9.0更改成v8.0)

再點擊上圖中的Path兩下,並移動下圖之環境變數位置。

(將CUDA v8.0的環境變數路徑都移到v9.0之上)

再開啟命令提示字元(cmd),並輸入 : nvcc --version 。

可以發現版本已經從v9.0更改成v8.0!!!

arrow
arrow
    文章標籤
    Caffe Tensorflow CUDA GPU
    全站熱搜

    水面上的小草 發表在 痞客邦 留言(0) 人氣()