web-dev-qa-db-fra.com

Comment gérer différentes versions des pilotes CUDA, CuDNN et Nvidia?

J'ai récemment mis à jour le GPU sur mon PC, sous Ubuntu 14.04

J'essaie de faire quelques tests avec Caffe , mais ils dépendront fortement de la version des pilotes CUDA, CuDNN et Nvidia.

Existe-t-il une option permettant de gérer différentes versions des pilotes CUDA, CuDNN et Nvidia?

Est-ce que différentes versions vont interférer les unes avec les autres? Est-il possible de changer de version?

Pilotes Nvidia installés:

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
14.04cudagpugpu-drivers
2
mrgloom

Je ne peux que répondre à une partie de votre question.

J'ai écrit un script pour basculer entre plusieurs versions de CUDNN. Il supporte actuellement les versions 5.1, 6.0 et 7.0. Vous pouvez trouver le script ici .

Une fois l'installation et le téléchargement des versions de CUDNN que vous souhaitez utiliser, vous pouvez basculer entre la v5.1, la v6.0 et la v7.0 en cliquant sur un bouton.

Voici ce que fait le script:

#!/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

Il effectue les opérations d'installation automatiquement pour la version dont vous avez besoin.

1
dnzzcn