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
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 :
# ...
sudo service docker restart
sudo docker run --gpus all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark