☰

如何将 Tensorflow 升级到 v1.3（cuDNN 和 CUDA 升级）

tag-icon • tag-icon cuda
如何将 Tensorflow 升级到 v1.3（cuDNN 和 CUDA 升级）

首先，下面是我的系统环境

  1. 操作系统：Ubuntu 16.04
  2. 当前 TensorFlow：1.0.1
  3. 当前 cuDNN：5.1.5
  4. 当前 CUDA：8.0

我想将 tensorflow 升级到 1.3，但是 tf 1.3 需要 cuDNN 6。

因此升级 cuDNN 是必要的，但我找不到方法升级cuDNN 和配置tensorflow 和我之前安装的一样。

答案1

我编写了一个脚本，可用于全新安装所需版本的 cuDNN 或升级现有版本。执行 GitHub 页面上的步骤后，您可以将 cuDNN 版本更改为 v6.0。之后，您无需对 TensorFlow 进行任何配置，它将自动开始使用 cuDNN v6.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

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

相关内容