Skip to content

Installation Guide

Jump to bottom
Jacky Liu edited this page Apr 10, 2018 · 1 revision

Welcome to the VINet wiki!

Docker installation

HOST setting

  • GPU: Nvidia GTX 1080
  • OS: Ubuntu16.04
  • CUDA 8.0 (driver 384.111)
sudo apt-get update
sudo apt-get install curl
curl -sSL https://get.docker.com | sudo sh
sudo usermod -aG docker $USER
sudo groupadd docker
sudo gpasswd -a $USER docker

Verify that Docker CE or Docker EE is installed correctly by running the hello-world image.

sudo docker run hello-world

NvidiaDocker installation

https://github.com/NVIDIA/nvidia-docker (Update: 2017/11/06)

# 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

# Test nvidia-smi
sudo nvidia-docker run --rm nvidia/cuda nvidia-smi

Download VINet docker image

(From DockerHub)

sudo docker login
sudo docker pull jjkka132/vinet

Start container

sudo nvidia-docker run --shm-size 8G --restart=always -d --name vinet -e PASSWORD='vinet' -p 192.168.0.1:8888:8888 -p 192.168.0.1:6006:6006 -p 192.168.0.1:52022:22 -v /home/${USER}/dockerData:/notebooks/data:z jjkka132/vinet:latest /bin/sh -c "while true; do sleep 10; done"

All related code in been put at /notebooks folder

Clone this wiki locally