Jak nainstalovat Cudnn na Ubuntu 16.04?

Pro TensorFlow bych chtěl nainstalovat cuda a CuDNN. Jak to udělám na Ubuntu 16.04?

Nejlepší odpověď

Krok 0:Nainstalujte cuda ze standardních úložišť. (Viz Jak mohu nainstalovat CUDA na Ubuntu 16.04?)

Krok 1:Zaregistrujte si vývojářský účet nvidia a stáhněte si cudnn zde (asi 80 MB)

Krok 2:Zkontrolujte, kde je vaše instalace cuda. Pro instalaci z úložiště je to /usr/lib/... a /usr/include . Jinak to bude /usr/local/cuda/ nebo /usr/local/cuda-<version> . Můžete to zkontrolovat pomocí which nvcc nebo ldconfig -p | grep cuda

Krok 3:Zkopírujte soubory:

Instalace úložiště:

$ cd folder/extracted/contents
$ sudo cp -P include/cudnn.h /usr/include
$ sudo cp -P lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
$ sudo chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*

Instalace Runfile:

$ cd folder/extracted/contents
$ sudo cp include/cudnn.h /usr/local/cuda/include
$ sudo cp lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*