為了讓Caffe和Tensorflow在windows中共存,所以CUDA要進行版本切換,因為Caffe需要使用CUDA 8.0,Tensorflow-gpu使用CUDA9.0,但安裝了CUDA 8.0和CUDA 9.0之後,Caffe在執行build時,Caffe的Cmakefile會自動偵測到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!!!
文章標籤
全站熱搜
留言列表