GPU Instance - KU-BIG/KUBIG_Wiki GitHub Wiki

hyeon95y/kubig:gpu

• hyeon95y/kubig:cpu 에 비해 이미지 사이즈가 크기 때문에, 충분한 용량의 인스턴스를 만들 것을 권장합니다
• 현재 테스트 중이기 때문에 발견되지 않은 이슈가 있을수 있습니다
  • ipywidgets 제대로 동작하지 않음

nvidia-docker2 설치

• 링크 참고
• How to setup Docker and Nvidia-Docker 2.0 on Ubuntu 18.04

$ sudo apt-get update
$ sudo apt-get upgrade
$ sudo apt-get install apt-transport-https ca-certificates curl software-properties-common
$ curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
$ 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 docker-ce

sudo 없이 docker를 사용하기 위해

$ sudo usermod -aG docker [Username]

예시)

$ sudo usermod -aG docker hyeon95y

$ curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
 sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
 sudo tee /etc/apt/sources.list.d/nvidia-docker.list
    
$ sudo apt-get update

    
$ sudo apt-get install -y nvidia-docker2
$ sudo pkill -SIGHUP dockerd

nvidia-driver-410 설치

• 링크 참고
• sudo apt install nvidia-410이 아닌
• sudo apt install nvidia-driver-410 으로 해야 설치됨

Run kubig:gpu Container

$ sudo docker run --runtime=nvidia -it --rm -p 9999:9999 -v ~:/home hyeon95y/kubig:gpu10.1

References

How to setup Docker and Nvidia-Docker 2.0 on Ubuntu 18.04

After pip 10 upgrade on pyenv "ImportError: cannot import name 'main'"

---

Run nvidia Container

• 이미지를 새로 만들고 싶을때, 이 이미지로부터 설치하여 commit

  $ sudo docker run --runtime=nvidia -it --rm nvidia/cuda:9.0-base