本文介绍了使用旧版本CUDA和cuDNN安装gpu版本的tensorflow的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
当前,我需要在已存在较旧版本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的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持!