Skip to content

L706077/Docker-CUDA-Caffe-install

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 

Repository files navigation

Docker-CUDA-Caffe-install


(1). Installer docker

$ sudo apt-get remove docker docker-engine docker.io
$ sudo apt-get update
$ sudo apt-get install -y apt-transport-https ca-certificates curl software-properties-common
$ curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
$ sudo apt-key fingerprint 0EBFCD88 
$ sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"
$ sudo apt-get update
$ sudo apt-get install -y docker-ce
$ sudo apt-get install docker-ce     ###开始安装docker 
$ apt-cache madison docker-ce        ###加载docker 
$ sudo docker run hello-world

Test whether install successfully or not

$ docker info 

Delete docker

$ sudo apt-get purge docker-ce
$ sudo rm -rf /var/lib/docker

show docker images

$ docker images

(2). Install nvidia-docker and nvidia-docker-plugin

$ wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb  
$ sudo dpkg -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb
$ sudo service nvidia-docker start
$ sudo nvidia-docker-plugin

(3). Install docker CUDA

$ nvidia-docker pull nvidia/cuda:9.0-runtime-ubuntu16.04
$ nvidia-docker run --rm nvidia/cuda:9.0-runtime-ubuntu16.04 nvidia-smi

than you can see the result as same as "$ nvidia-smi"

(4). Install docker Caffe

$ nvidia-docker run -ti bvlc/caffe:gpu caffe --version 

run caffe example

$ sudo nvidia-docker run -t -i bvlc/caffe:gpu /bin/bash 

or 
$ sudo nvidia-docker run -it bvlc/caffe:gpu
$ sudo nvidia-docker run -ti caffe:gpu bash -c "cd /opt/caffe/build; make runtest"
$ cd /opt/caffe
$ sh data/mnist/get_mnist.sh
$ sh example/mnist/create_mnist.sh
$ sh example/mnist/train_lenet.sh

share local data to container

$ sudo docker run -it -v /home/ubuntu/workspace/XXX:/share bvlc/caffe:gpu /bin/bash

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published