项目场景:
我需要跑通一个两年前的深度学习开源代码,作者给出的环境是:
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不匹配ÿ