Enregistrement de travail d'installation de tensorflow, CUDA, cuDNN sur Ubuntu 20.04 édition cuDNN

Contenu de cet article

J'ai enregistré le travail lorsque CUDA et cuDNN correspondant à la version de tensorflow ont été installés localement pour l'apprentissage automatique avec tensorflow.

Cet article a configuré CUDA 10.1 et cuDNN 7.6 pour utiliser tensorflow 2.3.0.

environnement

Ubuntu 20.04

Site de référence

https://codelabo.com/posts/20200229081221
https://medium.com/@exesse/cuda-10-1-installation-on-ubuntu-18-04-lts-d04f89287130

Structure de cet article

--1: installation du pilote NDIVIA -2: Configuration CUDA --3: Configuration cuDNN

Configuration cuDNN

Site de téléchargement: https://developer.nvidia.com/cudnn L'enregistrement d'un compte Nvidia est requis à l'avance. Vous pouvez sélectionner la version dans les archives du site de téléchargement.

1. Sélection de cuDNN 7.6

• Download cuDNN v7.6.5 (November 5th, 2019), for CUDA 10.1
• cuDNN Library for Linux

2. Installation à partir du fichier tar

Reportez-vous au Guide officiel et exécutez-le dans l'ordre.

2.1. Décompression

$ tar -zxvf cudnn-10.1-linux-x64-v7.6.5.32.tgz

2.2. Copier cudnn dans le cuda pré-installé

Copiez cudnn dans cuda / ou cuda- <version> /. La documentation officielle dit cuda /, mais il semble que cuda- <version> / et cuda / sont liés, et la copie vers cuda- <version> / a bien fonctionné.

# cuda-<version>/Lors de la copie vers
$ sudo cp cuda/include/cudnn*.h /usr/local/cuda-<version>/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-<version>/lib64
$ sudo chmod a+r /usr/local/cuda-<version>/include/cudnn*.h /usr/local/cuda-<version>/lib64/libcudnn*

# cuda/Lors de la copie vers
$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

Veuillez mettre la version de CUDA que vous avez mise dans <version>.

Lors du changement de version de cuda et cudnn en raison du changement de version de tensorflow

Installez la version souhaitée de cuda et modifiez ou ajoutez le chemin de cuda

Installez CUDA à l'avance. Veuillez vous référer à Configuration CUDA.

Si vous l'ajoutez, il semble qu'il sélectionnera automatiquement la version optimale de CUDA PATH et l'exécutera.

Référence: https://qiita.com/takeajioka/items/8737fab5cffbe0118fea

En fait, il existe de nombreuses façons de le faire. Modifiez le chemin décrit dans ~ / .bashrc.

Exemple

<Avant le changement 10.1>
export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
<Après le changement 11.0>
export PATH=/usr/local/cuda-11.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

changements cudnn

Lors de la copie de cudnn dans cuda- <version> /

Si vous changez la version de cuda chargée sur votre PC, il passera automatiquement à la version de cudnn qui est copiée sur le cuda modifié. (Je ne l'ai pas encore fait)

Lors de la copie de cudnn dans cuda /

/usr/local/cuda/include  
/usr/local/cuda/lib64  

Puisque l'ancien cudnn est inclus dans, supprimez-le et copiez la version nouvellement téléchargée de cudnn de la même manière.

$ sudo rm /usr/local/cuda/include/cudnn*.h
$ sudo rm /usr/local/cuda/lib64/libcudnn*

Si vous le copiez dans cuda /, vous devrez le supprimer lorsque vous changez le cuda, il peut donc être plus pratique de le copier dans cuda- <version> et de changer le cuda entier.