Richten Sie die Arbeitsaufzeichnung von Tensorflow, CUDA, cuDNN unter Ubuntu 20.04 cuDNN Edition ein

Inhalt dieses Artikels

Ich habe die Arbeit aufgezeichnet, als CUDA und cuDNN, die der Version von Tensorflow entsprechen, lokal für maschinelles Lernen mit Tensorflow installiert wurden.

In diesem Artikel wurden CUDA 10.1 und cuDNN 7.6 für die Verwendung von Tensorflow 2.3.0 eingerichtet.

Umgebung

Ubuntu 20.04

Referenzseite

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

Struktur dieses Artikels

--1: Installation des NDIVIA-Treibers -2: CUDA-Setup --3: cuDNN-Setup

cuDNN-Setup

Website herunterladen: https://developer.nvidia.com/cudnn Die Registrierung eines Nvidia-Kontos ist im Voraus erforderlich. Sie können die Version aus dem Archiv der Download-Site auswählen.

1. Auswahl von cuDNN 7.6

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

2. Setup aus der TAR-Datei

Lesen Sie den Offiziellen Leitfaden und führen Sie ihn nacheinander aus.

2.1 Dekompression

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

2.2 Kopieren Sie cudnn auf das vorinstallierte cuda

Kopieren Sie cudnn entweder nach cuda / oder cuda- <Version> /. In der offiziellen Dokumentation steht "cuda /", aber es scheint, dass "cuda- /" und "cuda /" verknüpft sind und das Kopieren in "cuda- /" gut funktioniert hat.

# cuda-<version>/Beim Kopieren nach
$ 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/Beim Kopieren nach
$ 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*

Bitte geben Sie die Version von CUDA ein, die Sie in "" eingegeben haben.

Beim Ändern der Version von cuda und cudnn aufgrund der Änderung der Version von tensorflow

Installieren Sie die gewünschte Version von cuda und ändern oder fügen Sie den cuda-Pfad hinzu

Installieren Sie CUDA im Voraus. Weitere Informationen finden Sie unter CUDA-Setup.

Wenn Sie es hinzufügen, scheint es, dass es automatisch die optimale CUDA-Version PATH auswählt und ausführt.

Referenz: https://qiita.com/takeajioka/items/8737fab5cffbe0118fea

In der Tat gibt es viele Möglichkeiten, dies zu tun. Ändern Sie den in ~ / .bashrc beschriebenen Pfad.

Beispiel

<Vor dem Wechsel 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}}
<Nach Änderung 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}}

cudnn Änderungen

Beim Kopieren von cudnn nach cuda- <Version> /

Wenn Sie die auf Ihrem PC geladene Version von cuda wechseln, wird automatisch auf die Version von cudnn umgeschaltet, die auf die geänderte cuda kopiert wird. (Ich habe es noch nicht getan)

Beim Kopieren von cudnn nach cuda /

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

Da das alte cudnn enthalten ist, löschen Sie es und kopieren Sie die neu heruntergeladene Version von cudnn auf die gleiche Weise.

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

Wenn Sie es nach "cuda /" kopieren, müssen Sie es löschen, wenn Sie das cuda ändern. Daher ist es möglicherweise bequemer, es nach "cuda- " zu kopieren und das gesamte cuda zu ändern.