docker, nvidia-docker install:
dockerhub登録後、docker login
で入れる。
sudo docker login
Username: hogehoge
・cuda使いたいならdevel版のimageを使う。
baseやruntimeだとnvccが通らない。
ここではダウンロードとかはしなくて良い。
FROMでイメージ名指定すれば持ってこれる。
sudo docker build -t hoge/yolov3_gpu -f Dockerfile .
runでコンテナが作成出来る。あんまり連発すると後でコンテナ消すのが面倒。
sudo docker run --runtime=nvidia -it -v `pwd`/temp:/root/work/temp hoge/yolov3_gpu bash
ここを自動で出来るようにしたい
・GPU=1, cuDNN=1にしてからmake
・cfgの設定(バッチサイズとか)変更
・exitでコンテナが停止する。内部データ等が削除されるわけではない。
・処理を止めたくないならdettachしてログインを抜ける。
"control+P、Q"でdettach出来る。
・dettach後や、起動中のコンテナに入りたい時はattachで入れる。
sudo docker attach docker-id