Ubuntu14.04とCUDA8.0でCaffeを動かすって話。
こんにちは、たっきーです。
なぜか前日までうんともすんとも動かなかったCaffeが今日はすんなり動いた。
理由は僕もわからない。
とりあえずUbuntu14.04にして、CUDA8.0にしてみたら動いた。
背景・概要
先日に引き続き「初めてのディープラーニング --オープンソース"Caffe"による演習付き」武井 宏将 (著)って本を参考に深層学習の勉強がしたい。
しかし、初期設定(?)で躓く。
バージョンを本の内容と合わせると動いた。
ようやくゴールが見えてきた。
先生曰く、「これ、2日でおわるからやってみてww」って感じだったんだけど僕はここまでくるのに雑魚なので2週間ぐらいかかってる。
初めてのディープラーニング --オープンソース"Caffe"による演習付き
- 作者: 武井宏将
- 出版社/メーカー: リックテレコム
- 発売日: 2016/02/19
- メディア: 単行本(ソフトカバー)
- この商品を含むブログ (8件) を見る
Ubuntu14.04インストール
「Ubuntu初心者がUbuntuのダウングレードしたって話」で書いたように、本の内容をトレースするようにUbuntuのバージョンを14.04にダウングレードした。下準備はOK。
CUDA7.5のインストール
CUDAのバージョンを7.5に合わせる方向性で行こうか。
とりあえず、CUDA7.5のダウンロードサイトへ。
$ sudo apt-get update$ sudo apt-get install cuda
これでCUDA7.5は無事インストール完了。
Caffeのインストール
以下、コマンドを端末で叩く。
$ sudo apt-get install libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev liblmdb-dev protobuf-compiler git
それぞれのライブラリの詳細はググろう。僕もあまりなんのためのライブラリなのか把握していない。
いよいよCaffeのダウンロード。
以下のコマンドを叩く。
$ git clone https://github.com/BVLC/Caffe.git
そろそろCUIの操作にも慣れてきたかもしれん。
g++-4.6のインストール
インストールコマンド
$ sudo apt-get install g++-4.6
Makeファイルの作成
Caffeのディレクトリまで移動する。
僕は /home にダウンロードしたのでそのまま
$ cd Caffe
で移動する。
geditコマンドでMakefile.configを編集する
geditコマンドでMakefile.configを開く。
$ gedit Makefile.config
# CUSTOM_CXX := g++の部分のコメントアウト(#)を外し、
CUSTOM_CXX := g++-4.6に変更する。
これでコンパイラの設定はOKかな。
保存を押して閉じる。
Caffeのコンパイル&テスト
Caffeのディレクトリにいることを確認して以下コマンドを叩く。
$ make
うまく行ったと思ったらエラー。
ちょっと調べる。
え。CUDA7.5はGTX1070をサポートしていない…???
参考にしたところはこちら。
CUDA 7.5 is old so it may not support 'compute_61'. CUDA8.0 or selecting 'compute_50' in Make.config will be a solution, maybe. I'm using CUDA8.0 + Geforce1070 with no problem.
英語読めないけどなんとなくGTX1070はCUDA7.5じゃだめだよーって書いてある気がする。
CUDA8.0にしたらいいよーって書いてある気がする。
本のバージョンとは異なるがGPUモードで計算したいので仕方なくCUDAのアップグレードする。
まずはUbuntu14.04の再インストール
これはなんとなくキレイにしたいから。
仕方ないからCUDA8.0をインストール
CUDA8.0のダウンロードページに行く。
7.5のときとは大体同じ手順。
必要なライブラリやCaffeのダウンロード・インストールは同じ手順で行う。
Caffeの再単体テスト
以下、そのコマンド(Caffeのディレクトリ内で行う。)
$ make$ make test$ make runtest
[ PASSED ]って表示されたら正常に走っているらしい。
やったぜ。
まとめ
英語読めるようになりたいね。
自分で読んでてこのブログ読み辛いね。。。
もっと簡潔に書けるように努力しよう。
初めてでも使える! Linux超入門 Ubuntu 16.10対応版 (日経BPパソコンベストムック)
- 作者: 日経Linux
- 出版社/メーカー: 日経BP社
- 発売日: 2016/12/13
- メディア: 単行本
- この商品を含むブログを見る
その日のディナー
今日のディナーです。 pic.twitter.com/a2uhk8eRGU
— 生産性のないツイートしかしません (@tacky0612) 2017年4月3日