C'est ce que j'ai fait. Il y a peut-être des choses supplémentaires que j'aurais pu ne pas avoir à faire, mais je vais quand même les inclure.
Tout d'abord, récupérez les pilotes du dépôt ppa. (Je dirais que cela est nécessaire avant l'installation, à moins que vous ne vouliez risquer une sorte de boucle de la mort).
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
Puis installez le dernier pilote. J'utilise la plupart du temps l'outil de mise à jour de l'interface graphique appelé Software & Updates, dans l'onglet Additional Drivers. A partir d'aujourd'hui, le pilote nvidia 396 est disponible.
Obtenez g++-6 et gcc-6. (Obligatoire)
sudo apt install g++-6
sudo apt install gcc-6
Vous pouvez essayer d'utiliser nvidia-cuda-toolkit, mais les chemins vers les bibliothèques ne me sont pas familiers. Je n'ai pas voulu m'en occuper.
(Je sauterais probablement celui-ci, mais allez-y si d'autres choses posent problème)
sudo apt install nvidia-cuda-toolkit
J'ai fini par installer le fichier d'exécution 9.1 pour Ubuntu 17.04. Téléchargez-le. Marquez le fichier comme exécutable (je fais un clic droit sur le fichier sur le bureau). Allez dans le terminal et mettez. (Nécessaire)
./cuda_9.1.85_387.26_linux.run --override
Il l'installera en utilisant les nouveaux compilateurs gcc. Il vous posera un tas de questions, et il voudra y répondre immédiatement.
Répondez oui à la configuration non prise en charge.
Non au pilote nvidia.
Oui - à la boîte à outils
J'ai utilisé les emplacements d'installation par défaut
Une fois que cela est installé, assurez-vous de configurer vos chemins. Le fichier d'exécution vous donnera un rappel, aussi. La documentation nvidia vous dira également ce qu'il faut exporter.
echo 'export PATH=/usr/local/cuda-9.1/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64:$LD_LIBRARY_PATH' >> ~/.bash.rc
source ~/.bashrc
Enfin, vous devez configurer les simlinks pour gcc-6 et g++-6, sinon vous obtiendrez un avertissement lors de la compilation de votre propre code (obligatoire).
sudo ln -s /usr/bin/gcc-6 /usr/local/cuda/bin/gcc
sudo ln -s /usr/bin/g++-6 /usr/local/cuda/bin/g++
Redémarrez le système. (Obligatoire)