Docker for A.I. Researcher

"Docker for A.I. Researcher" is a series of Shell script that
- allows you to quickly set up your deep learning research environment.
- supports the next-generation pytorch library, pytorch-lightning.
- supports SSH! Remote work with laptop OUTSIDE of the box.
- includes fancy terminal setup with oh-my-zsh.
Demo
Docker: Try it on Binder
1. Requirements
- Nvidia GPU Driver Installation
- 10 minuites
2. What's inside?
├── scripts
│ ├── 1-basic_install.sh
│ ├── 2-docker_install.sh
│ ├── 3-docker_build.sh
│ ├── 4-run_docker.sh
│ └── 5-setting_alias.sh
- 1-basic_install.sh : Install NVIDIA Driver + Terminal Utilities
- 2-docker_install.sh : Install Docker + NvidiaDocker2
- 3-docker_build.sh : Setup Environment. Dockerfile + Docker Image build.
- 4-run_docker.sh : Run Docker Container.
- 5-setting_alias.sh : Optional. Setting some happy alias.
3. Quick Start
step 1. clone the repository
sudo apt-get install git
git clone https://github.com/eungbean/Docker-for-AI-Researcher
cd Docker-for-AI-Researcher
Step 2. Install NVIDIA Driver and Terminal setup.
sudo sh scripts/1-basic_install.sh
Install following packages.
- NVidia Driver
- Terminal tools
- zsh
- oh-my-zsh
- zsh-syntax-highlighting
- zsh-autosuggestions
- neovim
- spacevim
- powerline font
- GPU Monitoring tools
- git
- openssh
Step 3. Install Docker
sudo sh scripts/2-docker_install.sh
exact same procedure from Nvidia Docker 2 Installation Guide.
Step 4. Configure Your Environment
configure your folder, port, etc.
Step 5. Build and run your Docker Container
sudo sh scripts/3-docker_build.sh
sudo sh scripts/4-run_docker.sh
- Initial root user id/password is
root:root.
- please change it using
passwd command inside the containder (don't worry. we'll do this together below).
- SSH to container will be automatically set up: Try below!
Step 6. Post Installation
1) Send ssh key to container
- SSH to container will be automatically set up: Try
ssh -p 10022 root@localhost and type password root.
2) Get inside docker container
- Initial ssh id/pw is
root/root.
- if you open 10022 port to the world, then you can access to your container via 10022 port from your laptop.
docker exec -it ${CONTAINER_NAME} /usr/bin/zsh
3) Change ssh password
(inside docker container)
passwd
You have to set your own password.