项目场景:
我需要跑通一个两年前的深度学习开源代码,作者给出的环境是:
TensorFlow 1.10 + CUDA 9.2 + CuDNNv7.5
但是我现在使用的是CUDA 11.2,因此需要在非管理员账号中自行安装对应版本的CUDA和CuDNN后,才能安装TensorFlow 1.10。
这里推荐博客:非root用户在linux下安装多个版本的CUDA和cuDNN(cuda 8、cuda 10.1 等)_随性拂尘倾心的博客-CSDN博客
问题描述
在依照上述博客安装完CUDA和CuDNN后,我选择pip+清华镜像源安装Tensorflow。在安装完成后的验证阶段,进入Python命令行,输入:
import tensorflow
给出报错信息:
原因分析:
发现是作者给出的CUDA和CuDNN版本号与TensorFlow 1.10不匹配,因为如下图所示,版本更高的tensorflow_gpu-1.12.0所搭配的CUDA和CuDNN的版本都比源代码作者给出配置,即CUDA 9.2 + CuDNNv7.5更低,因此需要重新安装CUDA和CuDNN。
解决方案:
重新安装CUDA和CuDNN,分别选择CUDA 9.0 + CuDNNv7.1,安装完后在Python交互界面输入
import tensorflow
后得到:
表明安装成功了!