ubuntu18.04安装tensorflow-gpu==1.14.0(CUDA10.0+cuDNN7.4.1)

hillyfighting

于 2020-07-05 09:26:37 发布

阅读量3.7k

点赞数 1

分类专栏：安装记录/环境配置 tensorflow

本文链接：https://blog.youkuaiyun.com/hillyfighting/article/details/106633362

版权

本文详细介绍了在Ubuntu18.04上安装TensorFlow-GPU 1.14.0的步骤，包括前提检查、Anaconda安装、CUDA10.0的安装、cuDNN7.4.1的安装，以及如何确保所有组件的版本匹配和正确安装。文章提供了解决GCC版本过高、环境变量配置、驱动与CUDA版本关联等问题的方法，并附有测试安装成功的步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

系统版本：ubuntu 18.04
系统架构：x86_64
nvidia driver 驱动版本：440.59
CUDA版本：10.0
cuDNN：7.4.1
gcc:4.8.5
tensorflow-gpu:1.14.0

前提

step1:确定是否安装了驱动:nvidia-smi
step2:查看gcc版本:gcc --version
若gcc版本太高就更改版本：sudo apt-get install gcc-4.8 g++-4.8
切换gcc版本：sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 100
激活生效：sudo update-alternatives --config gcc
step3:确定GPU卡可以进行CUDA计算，并且计算力>3.5
到nvidia官网查
step4:找好对应的版本：
tensorflow版本<—>CUDA版本<—>cuDNN版本。tensorflow,CUDA,cuDNN之间有严格依赖关系，对应版本一定要对应好。
驱动版本与CUDA有关联，CUDA要求驱动大于某个版本。