web-dev-qa-db-fra.com

Tensorflow 1.12 est-il compatible avec CUDA 10.1?

J'ai réussi à configurer un serveur Ubuntu 18.04 avec nvidia-smi 418.39, Driver version 418.39 et CUDA 10.1

J'ai maintenant un utilisateur qui veut exécuter TensorFlow mais insiste sur le fait qu'il n'est pas compatible avec CUDA 10.1, seulement CUDA 10. Il n'y a aucune déclaration confirmant cela en ligne où que je puisse trouver, ni dans les notes de mise à jour de TF. Parce que la mise en place de ce système était un peu difficile à faire, j'hésite un peu à essayer de rétrograder une seule version.

Quelqu'un peut-il vérifier si TensorFlow 1.12 fonctionne ou non avec CUDA 10.1?

12
Eric Berry

Je peux également confirmer que tf 1.13.1 ne fonctionne pas avec CUDA 10.1. Lors de l'importation de tensorflow, vous obtiendrez l'erreur suivante

ImportError: libcublas.so.10.0: impossible d'ouvrir le fichier d'objet partagé: aucun fichier ou répertoire de ce type

l'exécution de ldconfig -v montre la différence entre libcublas.so.10.0 et libcublas.so.10.1.0.105

1
Fisa