Prepare a development environment for executing docker image with GPU support

1. Install Docker Engine [1]

$ sudo apt-get remove docker docker-engine docker.io containerd runc

$ sudo apt-get update 

$ sudo apt-get install apt-transport-https ca-certificates curl gnupg-agent 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 docker-ce-cli containerd.io

2. Add your account into docker group

$ sudo gpasswd -a $USER docker

$ sudo reboot

3. Install nvidia-container-runtime [2]

$ curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | sudo apt-key add -

$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

$ curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list

$ sudo apt update

$ sudo apt install nvidia-container-runtime

$ sudo systemctl restart docker.service

4. Check

$ docker run -it --rm --gpus all ubuntu nvidia-smi


 

Reference:

[1] https://docs.docker.com/engine/install/ubuntu/

[2] https://github.com/NVIDIA/nvidia-container-runtime

留言

這個網誌中的熱門文章

Tuing PID parameters in QGroundcontrol (2)

Useful PX4 Parameters

Burn linux image to eMMC storage on Banana Pi M3