Ubuntu16.04+CUDA9.0+CUDNN7.1+Tensorflow-gpu-1.11.0详细安装教程

最新推荐文章于 2024-08-14 09:08:25 发布

同学醒醒放学了

最新推荐文章于 2024-08-14 09:08:25 发布

阅读量2.6k

点赞数 1

文章标签： tensorflow

本文链接：https://blog.csdn.net/weixin_41399111/article/details/84898307

版权

本文提供Ubuntu16.04系统下Tensorflow-gpu-1.11.0的详细安装教程。首先确保系统已安装pip，然后根据Tensorflow版本选择对应的CUDA和CUDNN。安装CUDA时避免安装NVIDIA驱动，以免系统崩溃。安装CUDNN后配置环境变量，使用清华源加速pip安装Tensorflow-gpu。最后验证安装成功，解决可能的库文件缺失问题。

摘要由CSDN通过智能技术生成

Ubuntu16.04+CUDA9.0+CUDNN7.1+Tensorflow-gpu-1.11.0安装教程

本文主要介绍在ubuntu16.04系统环境下进行Tensorflow-gpu的安装，本文的安装方式是通过pip 工具进行安装，因此在安装TF前要保证系统有pip工具，假如没有按照如下指令安装：

# 1. 更新系统包
sudo apt-get update
sudo apt-get upgrade

# 2. 安装Pip
sudo apt-get install python-pip   #若python为3.x，指令为 sudo apt-get install python3-pip

# 3. 检查 pip 是否安装成功
pip -V

安装完成后，这时候需要确定自己安装的Tensorflow-gpu版本，不同的版本对应的cuda以及cudnn也不同，当然tensorflow的CPU版本并不要求cuda.

版本依赖cuda及cudnn具体对照表如下：