☰

如何管理不同版本的 CUDA、CuDNN 和 Nvidia 驱动程序？

tag-icon • tag-icon 14.04 cuda gpu gpu-driver
如何管理不同版本的 CUDA、CuDNN 和 Nvidia 驱动程序？

我最近更新了运行 Ubuntu 14.04 的电脑中的 GPU

我正在尝试运行一些基准测试咖啡，但它们将严重依赖于 CUDA、CuDNN 和 Nvidia 驱动程序的版本。

有没有选项可以管理不同版本的 CUDA、CuDNN 和 Nvidia 驱动程序？

不同版本之间会互相干扰吗？可以切换版本吗？

已安装的 Nvidia 驱动程序：

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

答案1

我只能回答你问题的一部分。

我编写了一个脚本来在多个 cuDNN 版本之间切换。它目前支持 v5.1、v6.0 和 v7.0。您可以找到该脚本这里。

完成安装并下载您想要使用的 cuDNN 版本后，您只需单击按钮即可在 v5.1、v6.0 和 v7.0 之间切换。

脚本的作用如下：

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

它会自动执行您需要的版本的安装操作。

相关内容