Pour TensorFlow, je voudrais installer CUDA. Comment puis-je le faire sur Ubuntu 16.04 ?
Réponses
Trop de publicités?J'ai fait cela plusieurs fois, avec ou sans succès, perdant mon écran, venant ici - gagnant des idées - certaines librairies cuda ne sont pas dans le chemin, manquantes, non installées - la manière saine est juste d'installer les pilotes linux pour votre carte nvidia. https://medium.com/techlogs/install-the-right-nvidia-driver-for-cuda-in-ubuntu-2d9ade437dec et travailler sur nvidia-cuda docker images - base ou devel.
Faites un mappage volumique de votre dossier de code vers le conteneur - installez ce que vous voulez. Idem pour travailler avec keras ou tensorflow ou simplement avec de l'informatique pure. opencv
docker run --net=host --runtime=nvidia -it -v ~/coding:/coding nvidia/cuda : /bin/bash
Note TF est également livré avec son docker
J'ai testé ces étapes pour installer cuda pour plusieurs LTS, y compris 20.04 et 22.04 ... la plupart des réponses ici sont obsolètes et doivent être évitées.
Il est utile d'utiliser une version LTS d'Ubuntu, qui apparaît chaque année paire comme dans 16.04, 18.04, 20.04, 22.0, etc... parce que c'est une cible moins mouvante pour les propriétaires de pilotes ET le site de téléchargement de Cuda ne liste spécifiquement que les versions LTS.
Purgez d'abord tous les pilotes nvidia que vous avez pu installer.
sudo apt purge *nvidia* -y # this will auto give you nouveau graphics drivers
confirmez ce qui précède en énumérant tous les paquets connexes
dpkg -l | grep nvidia # do above purge if need be
Supprimer tous les fichiers dans le répertoire /var/crash
ou l'installation ci-dessous ne fonctionnera pas
reboot
Aller à la page de téléchargement de cuda https://developer.nvidia.com/cuda-downloads Linux -> x86_64 (sauf si vous êtes sur ARM) -> Ubuntu -> choisissez le LTS correspondant à votre installation Ubuntu.
Ensuite, dans un terminal, tapez toutes les commandes listées dans le site ci-dessus.
reboot
Maintenant, confirmez que vous avez installé cuda en émettant
nvcc --version
AVIS - ce qui précède installera automatiquement les pilotes graphiques nvidia appropriés pour correspondre à votre version Ubuntu LTS, la version cuda et votre matériel.
- Réponses précédentes
- Plus de réponses