從零開始學caffe(八):Caffe在Windows環境下GPU版本的安裝

之前我們已經安裝過caffe的CPU版本,但是在MNIST手寫數字識別中,我們發現caffe的CPU版本運行速度較慢,訓練效率不高。因此,在這裏我們安裝了caffe的GPU版本,並使用GPU版本的caffe同樣對手寫MNIST數字集進行訓練。 step1: 安裝CUDA 首先需要準備好NVIDIA的顯卡,然後下載安裝CUDA,在這裏我安裝的是CUDA8.0下載地址。CUDA的安裝過程比較簡單,在這
相關文章
相關標籤/搜索