Docker instellen om Nvidia GPU-compatibele container te gebruiken [Ubuntu 18.04]
Vanaf versie 19.03 ondersteunt Docker native "versnelling van containers met Nvidia GPU's".Deze functie wordt beschikbaar gesteld door gebruik te maken van de NVIDIA Container Toolkit.Maak hier de eerste instellingen die daarvoor nodig zijn.
環境
Besturingssysteem: Ubuntu 18.04 LTS
GPU: GTX1080
- 1. Stap XNUMX: Controleer de vereisten
- 2. Nvidia docker 2 installatieprocedure en bevestiging van vereisten [Ubuntu 18.04]
- 3. Stap XNUMX: Installeer Docker
- 4. Installeer de nieuwste versie van Docker CE op Ubuntu 18.04 [Document-compatibel]
- 5. Stap XNUMX: Installeer Nvidia GPU-stuurprogramma
- 6. Twee manieren om NVIDIA GPU-stuurprogramma te installeren [Ubuntu 18.04]
- 7. Stap XNUMX: Maak GPU-container beschikbaar in Docker
- 8. Controleer de werking
Stap XNUMX: Controleer de vereisten
Voordat we Docker's "versnelling van containers door GPU" gaan gebruiken, kijken we eerst naar de omgeving.GPU-containers kunnen niet worden gebruikt in omgevingen die te oud zijn.
- GNU / Linux x86_64 (kernelversie> 3.10) ← Ubuntu 18.04 dus OK
- NVIDIA GPU met architectuur> Fermi (2.1) ← Als GPU voor desktop is, is GTX 600 of hoger OK
- NVIDIA-stuurprogrammaversie ≧ 361.93 ← Zie stap XNUMX hieronder
Zie de volgende pagina voor details.
Stap XNUMX: Installeer Docker
Als je Docker nog niet hebt geïnstalleerd, volg je de onderstaande stappen om het te installeren.
Stap XNUMX: Installeer Nvidia GPU-stuurprogramma
Het Nvidia-stuurprogramma is vereist om de GPU-container (CUDA-container) uit te voeren.
De versie van de Nvidia-driver moet meer zijn dan de versie van CUDA die u wilt gebruiken (tensorflow heeft bijvoorbeeld exact de CUDA-versie nodig, afhankelijk van de versie die u gebruikt).
Zolang u de Nvidia-driver correct installeert, hoeft u de CUDA-toolkit of NVIDIA Container Toolkit niet afzonderlijk te installeren.
De installatiemethode van Nvidia Driver wordt hieronder samengevat.
Stap XNUMX: Maak GPU-container beschikbaar in Docker
Voer het volgende in vanaf de terminal.
Officiële referentie: https://github.com/NVIDIA/nvidia-docker
# Add pakket repository distributie = $ (. / 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 herstart docker
Controleer de werking
#Getest met de nieuwste officiële CUDA-image docker-run --gpus all nvidia / cuda: 9.0-base nvidia-smi
Als alles goed gaat, ziet u dit display.
Om de GPU-container vanaf nu te gebruiken, voegt u de "–gpus" -optie toe aan "docker run".
Als je zojuist docker hebt geïnstalleerd, wordt het mogelijk niet uitgevoerd, tenzij je sudo toevoegt voordat docker wordt uitgevoerd.
[Ubuntu] Instellingen en voorzorgsmaatregelen voor het uitvoeren van docker zonder het sudo-commando
Dit zijn de initiële instellingen die nodig zijn om Docker's GPU-compatibele container op Ubuntu te gebruiken.
Referentieboek