Configure o Docker para usar o contêiner compatível com GPU Nvidia [Ubuntu 18.04]
A partir da versão 19.03, o Docker oferece suporte nativo para "aceleração de contêineres com GPUs Nvidia".Este recurso é disponibilizado usando o NVIDIA Container Toolkit.Aqui, faça as configurações iniciais necessárias para isso.
Meio ambiente
SO: Ubuntu 18.04 LTS
GPU: GTX1080
- 1. Etapa XNUMX: verifique os pré-requisitos
- 2. Procedimento de instalação do Nvidia docker 2 e confirmação de pré-requisitos [Ubuntu 18.04]
- 3. Etapa XNUMX: instalar o Docker
- 4. Instale a versão mais recente do Docker CE no Ubuntu 18.04 [Document compliant]
- 5. Etapa XNUMX: Instale o driver GPU Nvidia
- 6. Duas maneiras de instalar o driver NVIDIA GPU [Ubuntu 18.04]
- 7. Etapa XNUMX: disponibilizar o contêiner de GPU no Docker
- 8. Verifique a operação
Etapa XNUMX: verifique os pré-requisitos
Antes de usar a "aceleração de contêineres por GPU" do Docker, vamos verificar o ambiente.Os contêineres de GPU não podem ser usados em ambientes muito antigos.
- GNU / Linux x86_64 (versão do kernel> 3.10) ← Ubuntu 18.04 então OK
- GPU NVIDIA com arquitetura> Fermi (2.1) ← Se a GPU for para desktop, GTX 600 ou posterior está OK
- Versão do driver NVIDIA ≧ 361.93 ← Veja a etapa XNUMX abaixo
Por favor, veja a seguinte página para detalhes.
Etapa XNUMX: instalar o Docker
Se você ainda não instalou o Docker, siga as etapas abaixo para instalá-lo.
Etapa XNUMX: Instale o driver GPU Nvidia
O driver Nvidia é necessário para executar o contêiner GPU (contêiner CUDA).
A versão do driver Nvidia deve ser mais do que a versão do CUDA que você deseja usar (por exemplo, tensorflow requer exatamente a versão CUDA dependendo da versão que você usa).
Contanto que você instale o driver Nvidia corretamente, você não precisa instalar o kit de ferramentas CUDA ou o kit de ferramentas de contêiner NVIDIA separadamente.
O método de instalação do driver Nvidia é resumido abaixo.
Etapa XNUMX: disponibilizar o contêiner de GPU no Docker
Digite o seguinte no terminal.
Referência oficial: https://github.com/NVIDIA/nvidia-docker
#Add package repository 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-container-toolkit sudo systemctl restart docker
Verifique a operação
#Testado com o último docker de imagem CUDA oficial executado --gpus all nvidia / cuda: nvidia-smi 9.0-base
Se tudo correr bem, você verá esta tela.
Para usar o contêiner da GPU a partir de agora, adicione a opção "–gpus" ao "docker run".
Se você acabou de instalar o docker, ele pode não ser executado a menos que você adicione o sudo antes da execução do docker.
[Ubuntu] Configurações e precauções para executar o docker sem o comando sudo
Estas são as configurações iniciais necessárias para usar o contêiner compatível com GPU do Docker no Ubuntu.
Livro de referência