使用旧版本CUDA和cuDNN安装gpu版本的tensorflow [英] Install gpu version tensorflow with older version CUDA and cuDNN

问题描述

当前，我需要在已存在较旧版本CUDA和cuDNN的计算机上安装gpu版本tensorflow，我的问题是如何在不重新安装CUDA和cuDNN的情况下完成它，因为其他软件包（请参阅pytorch）需要它。

解决方案

根据

Currently, I need to install gpu version tensorflow on a machine that existing an older version CUDA and cuDNN, and my question is how can I get it done without re-installing CUDA and cuDNN since other package (see pytorch) need it.

解决方案

Check CUDA version with below command according to this post.

$ nvcc --version

Check cuDNN version with below command according to this post.

$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

Choose tensorflow-gpu version according to this post

这篇关于使用旧版本CUDA和cuDNN安装gpu版本的tensorflow的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！