Skip to content

Instantly share code, notes, and snippets.

@abcinje

abcinje/image.md

Last active May 23, 2020 06:50
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save abcinje/8875f764c5fb2c08abe04579b77e9a32 to your computer and use it in GitHub Desktop.
Save abcinje/8875f764c5fb2c08abe04579b77e9a32 to your computer and use it in GitHub Desktop.
Download ZIP
bufferring image recipe
Raw
image.md
  1. Generate a container
$ docker run -it --name bufferring --network=host nvidia/cuda:10.2-devel /bin/bash
  1. Install packages
# apt update
# apt install -y build-essential git openssh-server python3 python3-pip vim wget
  1. Install torch, torchvision and tqdm
# pip3 install torch torchvision tqdm

  1. Install MPI [link]

  2. Install mpi4py

# pip3 install mpi4py

  1. Configure ssh
    In /etc/ssh/sshd_config, set PermitRootLogin to yes

  2. Change password

# passwd
  1. Start ssh server automatically
# echo "service ssh start" >> $HOME/.bashrc
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment