
Caffe 1.0をNVIDIA Dockerで試してみた
RCの取れたCaffe 1.0とそのDockerイメージがリリースされましたので、NVIDIA Docker上でResNet-50のベンチマークを実施してみました。
RCの取れたCaffe 1.0とそのDockerイメージがリリースされましたので、NVIDIA Docker上でResNet-50のベンチマークを実施してみました。
Docker CE 17.03に対応したNVIDIA Docker 1.0.1をUbuntu 16.04 LTSにインストールしてみました。
Docker CEをUbuntu 16.04 LTSにインストールしてみました。
Ubuntu 16.04 LTSに対応したNVIDIA Machine Learningリポジトリを追加して、apt-getでcuDNN 5.1をインストールしてみました。
CUDA 8.0とNVIDIA DriverをUbuntu 16.04 LTSにインストールしてみました。
Ubuntu 16.04.2とJapanese Teamのパッケージレポジトリの組み合わせで、Ubuntu 16.04 LTS環境を構築してみました。
Docker 1.13.1とnvidia-docker 1.0.0をそれぞれDocker 17.03.0とnvidia-docker 1.0.1に更新してみました。
GitHub版に続き、Docker版のNVIDIA DIGITS 5.0がDocker Hubでリリースされましたので、nvidia-dockerで試してみました。
Linuxプラットフォーム用「MXNet 0.9.3」がPyPIに登録されていたので、Dockerコンテナにインストールしてみました。まだ、Chainerのように、pipで簡単インストールとはいかないようです。
前回の記事「NVIDIA DIGITSでファインチューニングしてみた」で作成したalexnet_finetune.prototxtを例に、NVIDIA DIGITS用のプロトテキストについて説明します。