Como gerenciar diferentes versões dos drivers CUDA, CuDNN e Nvidia?

Question

Atualizei recentemente a GPU no meu PC, executando o Ubuntu 14.04

Estou tentando fazer alguns benchmarks com o Caffe , mas eles dependem muito da versão dos drivers CUDA, CuDNN e Nvidia.

Existe alguma opção para gerenciar diferentes versões dos drivers CUDA, CuDNN e Nvidia?

As diferentes versões interferirão entre si? É possível alternar entre versões?

Drivers da Nvidia instalados:

apt-cache search nvidia | grep version

nvidia-173 - NVIDIA legacy binary driver - version 173.14.39
nvidia-304 - NVIDIA legacy binary driver - version 304.131
nvidia-304-updates - NVIDIA legacy binary driver - version 304.131
nvidia-340 - NVIDIA binary driver - version 340.96
nvidia-340-updates - NVIDIA binary driver - version 340.96
nvidia-352-updates - NVIDIA binary driver - version 352.63
nvidia-352 - NVIDIA binary driver - version 352.79
nvidia-355 - NVIDIA binary driver - version 355.11
nvidia-358 - NVIDIA binary driver - version 358.16
nvidia-361 - NVIDIA binary driver - version 361.45.18
nvidia-364 - NVIDIA binary driver - version 364.19
nvidia-367 - NVIDIA binary driver - version 367.44
nvidia-370 - NVIDIA binary driver - version 370.23

    

Answer 1

Eu só posso responder uma parte da sua pergunta.

Eu escrevi um script para alternar entre várias versões do cuDNN. Atualmente suporta v5.1, v6.0 e v7.0. Você pode encontrar o script aqui .

Depois de concluir a instalação e baixar as versões do cuDNN que você deseja usar, você poderá alternar entre v5.1, v6.0 e v7.0 com o clique de um botão.

Isso é o que o script faz:

#!/bin/bash

rm -f /usr/include/cudnn.h
rm -f /usr/lib/x86_64-linux-gnu/*libcudnn*
rm -f /usr/local/cuda-*/lib64/*libcudnn*


cp -P packages/cudnn/include/cudnn.h /usr/include
cp -P packages/cudnn/lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*

rm -rf packages/cudnn

Realiza as operações de instalação automaticamente para a versão que você precisa.