Ubuntu18.0 下 Tensorflow-GPU安装

最新推荐文章于 2023-12-12 23:26:43 发布

愚人越

最新推荐文章于 2023-12-12 23:26:43 发布

阅读量393

点赞数 1

分类专栏： tensorflow

本文链接：https://blog.csdn.net/qq_40919254/article/details/85112399

版权

本文详细记录了在Ubuntu 18.04系统上安装Tensorflow-GPU的过程，包括确认GPU支持CUDA、检查系统信息、安装CUDA和cudnn、设置环境变量以及安装Tensorflow-GPU。针对可能出现的问题，如选择合适的CUDA和cudnn版本、处理驱动安装冲突等，给出了解决方案。

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

前言：因为要跑比赛，要用到GPU，所以决定安装tensorflow-gpu版本，首先感谢广大博客网友的无私奉献，当然每个人遇到的情况不一样，会有一些坑，因为显卡驱动和tensorflow一直在更新，所以不同版本的安装方式也不一样，本文配置如下

x86_64
Ubuntu 18.0
python2.7
GPU：Quadro k600 / Tesla k20c
cuda-10.0
cudnn runtime+cudnn developer
tensorflow-gpu-1.12.1

确认电脑显卡支持cuda

$ lspci | grep -i nvidia

此时你的电脑会显示显卡个数和型号

$ lspci | grep -i nvidia
02:00.0 VGA compatible controller: NVIDIA Corporation GK107GL [Quadro K600] (rev a1)
02:00.1 Audio device: NVIDIA Corporation GK107 HDMI Audio Controller (rev a1)
03:00.0 3D controller: NVIDIA Corporation GK110GL [Tesla K20c] (rev a1)

然后去NVIDIA官网查看对应型号及其性能，性能大于等于3.0的NVIDIA的显卡
（不知道自己的NVIDIA GPU的计算性能的可以在这里查到： https://developer.nvidia.com/cuda-gpus ）