沒有GPU沒關係,沒有nvidia顯卡也沒關係,基於docker可以3分鐘配好caffe!
Docker簡單來說就是一種輕量虛擬機,可以快速部署和執行程序。
之前看過許多配置caffe的資料,主要是硬件相關的配置因爲電腦個體差異會引發各種各樣的問題,後來遇到docker,可以直接pull別人配好的caffe環境,終於可以在自己電腦(ubuntu 15.04)歡脫地跑caffe了><
docker畢竟還是有自己的侷限,比如界面,比如數據共享等。如果不滿足於虛擬容器,需要在真機配置,出門左轉 caffe學習筆記3:ubuntu15.04 caffe配置
1 安裝docker
參考這裏的步驟“安裝docker”和“去除sudo”,其中的命令直接複製到終端執行。
需要注意的是:在“去除sudo”的步驟中,將youngboy改爲自己的用戶名即可。如果重啓後執行docker version看不到期望的效果,有可能是docker沒有開,運行sudo service docker start
2 下載caffe虛擬環境
在這裏有docker的caffe鏡像,有CPU和GPU兩個版本,將右側的“Docker pull command”中的命令docker pull tleyden5iwx/caffe-cpu-master
複製到shell中運行,下載已配好的caffe到自己的docker中。要下載docker鏡像需要在這個docker hub上註冊,在搜索欄可以查找自己需要的虛擬鏡像。
在shell中運行 docker run -i -t tleyden5iwx/caffe-cpu-master bash
將運行caffe虛擬鏡像,運行cat /etc/issue
可以查看這個鏡像的系統信息。
然後運行這裏提供的測試例程“Verify CPU version”驗證caffe是否可用。我的在運行時有“libdc1394 error”
再運行ln -s /dev/null /dev/raw1394
後,解決了問題,成功地跑通用MNIST訓練LeNet的例程。
3 實際使用
- 在docker虛擬鏡像中生成的數據需要用docker commit保存
- 可以在這個docker鏡像中配置SSH或配置FTP實現和本機的文件共享