우분투 18.04 에서 Docker 설치하기

회사에서 사용하는 기본 이미지 배포를 진행하기 위해 18.04에서 도커를 사용하고자 한다.

nvidia 그래픽 카드의 cuda를 사용하여 가끔 머신러닝 관련된 작업도 진행하기 때문에 도커에서 그래픽카드 사용이 가능한

nvidia-docker2를 설치하게 되었다. 

 

설치는 ubuntu18.04 , 당연히 도커가 구동 될 컴퓨터에 nvidia그래픽 카드가 필요하다.

 

설치를 위해 터미널에서 다음과 같이 입력한다. 

sudo apt update

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 bionic stable"

sudo apt update

sudo apt install docker-ce


distribution=$(. /etc/os-release;echo $ID$VERSION_ID) && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - && 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 systemctl restart docker

설치가 끝난 뒤 아래 코드를 터미널에서 입력해서 제대로 설치가 됬는지 확인한다.

sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi <- test

설치가 완료됬다면 위와 같이 터미널에서 nvidia-smi이 떠있을 것이다!