NVIDIA® CUDA® Toolkit¶
Installation avec WSL¶
- Installer le driver CUDA Toolkit sur la machine Windows (il ne faut pas l'installer dans la VM Ubuntu)
- cuda-toolkit/install-wsl.sh traite les étapes suivantes en contexte WSL :
curl -sSL https://mborne.github.io/outils/cuda-toolkit/install-wsl.sh | bash
Utilisation avec docker¶
cuda-toolkit/install-container-toolkit.sh ajoute le support GPU à Docker en installant nvidia-container-toolkit avec un redémarrage et un test de bon fonctionnement :
curl -sSL https://mborne.github.io/outils/cuda-toolkit/install-container-toolkit.sh | bash
Tester l'installation¶
Vous pouvez rejouer le test du script précédent :
sudo docker run --gpus all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark