天河二號執行CAFFE CIFAR10例程
阿新 • • 發佈：2018-11-19
天河二號自備有caffe1.0.0的程式庫，所以不需要自行編譯（如果有特殊需要也可以自行編譯）
執行caffe例程有以下幾個步驟：
- 載入caffe：
module load caffe/1.0.0這一步系統會自動載入對應的環境變數進入系統中
呼叫which caffe可以檢視caffe程式的具體位置。
2. 上傳對應的資料和proto檔案到相應的資料夾中。
CIFAR10資料需要自己下載，下載地址：
下載後，通過WinScp上傳到EXAMPLE/data
執行caffe自帶的
convert_cifar_data程式，在當前目錄為data資料夾下，執行指令convert_cifar_data cifar10 ouput，將原始bin資料檔案轉換成caffe能用的lmdb檔案。然後執行compute_image_mean -backend=lmdb output/cifar10_train_lmdb output_mean/mean.binaryproto 生成圖片的均值lmdb。 `
caffe設定檔案使用caffe原始碼中自帶的examples/cifar10中的例程，有兩個檔案是需要的，cifar10_full_train_test.prototxt
cifar10_full_solver_lr1.prototxt上傳到EXAMPLE/data資料夾內。並修改cifar10_full_train_test.prototxt中的資料來源為對應的資料來源，包括mean_file,以及source，注意有兩個空間需要修改，train和test都需要，如圖所示。 
最終目錄如圖所示
根目錄
data
proto
最終在EXAMPLE/目錄執行執行yhrun -p 你所在的分割槽代號 caffe train --solver=proto/cifar10_full_solver_lr1.prototxt啟動caffe程序，提交到超算平臺分配節點進行計算。