1. 시스템 사양
시스템 사양은 아래와 같습니다.
2. NVIDA CUDA GPU 설치
Ubuntu 20.04 에서는 Nvidia CUDA 드라이버가 설치 되어 있습니다. 하지만 CUDA 버전을 설치 하는 것이 결정되어 있지 않습니다. 그래서 아래와 같이 미리 GPU를 삭제 하여야 합니다.
$ sudo apt-get remove --purge '^nvidia-.*'
$ sudo apt-get autoremove --purge 'cuda*'
$sudo apt install nvidia-cuda-toolkit
$ nvcc -V
$ ubuntu-drivers devices
3. NVIDIA DOCKER 이미지 설치
향후 특정 사용자가 Docker 를 사용하게 하려면, 사용자를 추가 하여야 합니다.
NVIDA Docker 설치 사이트
위의 링크와같이 설치 하였다면, Docker 까지 같이 설치 되었을 것입니다.
4. Nvidia 이미지내 명령 실행
아래와 같이 명령어를 차례대로 수행 하시면, Docker 실행 되는 여부와 CUDA를 실행 하기 위한 Path도 지정 됩니다.
$ sudo nvidia-docker run --rm hello-world
$ apt-cache madison nvidia-docker2 nvidia-container-runtime
$ sudo docker run -it nvidia/cuda:11.0-base
5. Nvidia Docer image 실행
$ sudo docker run -it --rm --runtime=nvidia -v $(realpath ~/Dropbox):/tf/notebooks -p 8888:8888 tensorflow/tensorflow:2.4.3-gpu-jupyter