Como utilizar o Docker NVIDIA CUDA Toolkit


Última atualização em 25 de Agosto, 2021 por Vítor Fernandes.

Numa versão anterior (Ubuntu 18.04) apresentado aqui, incluía um conjunto de instruções e requisitos que já foram ultrapassados após a atualização do controlador NVIDIA WSL 2 471.21

1 – Ativa o Linux Subsytem (WLS 2) no Windows

Segue os passos apresentados aqui, depois instala o Ubuntu 20.04 LTS através da loja Microsoft

1.1 – Verifica a versão do kernel

O kernel terá que ser igual ou superior à versão 5.10.16.3

uname -r

2 – Instala o Docker com suporte WSL 2

Descarrega o ficheiro executável através da página oficial Docker

2.1 – Ativa a integração do Docker com o Ubuntu através do WSL 2

Acede ao dashboard, em definições. Seleciona Resources, depois WSL Integration e seleciona a imagem Linux

Docker com Ubunto através do WSL 2

3 – Adiciona o repositório e instala o controlador NVIDIA CUDA Toolkit 11.4

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda-repo-wsl-ubuntu-11-4-local_11.4.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-11-4-local_11.4.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-wsl-ubuntu-11-4-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

4 – Atribui a variável no sistema

export PATH=/usr/local/cuda-11.4/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64\
                         ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}