ubuntu14.04下tensorflow安装指南（GPU）

2023年11月5日 110次阅读 来源: Square Tsou

本文写于2017年7月，若多个月后才看到此文的不建议参考，因为定会有新坑出现

作者的GPU是NVIDIA 的TITAN X系列，支持CUDA加速，所以本文主要针对GPU版本的安装

因为作者的ubuntu是才安装的，没有安装过其他软件，所以安装过程是最基础的过程，默认没有安装过任何支持文件

以下为安装步骤

1.安装cuda

在此链接中 下载对应版本的cuda8.0，作者下载的是run的版本，文件名为cuda_8.0.61_375.26_linux.run

CUDA Toolkit Download(ci lian jie wu xiao)

Index of /compute/cuda/repos/ubuntu1404/x86_64

进入到刚才下载的cuda位置，更改权限

sudo chmod +x cuda_8.0.61_375.26_linux.run

按 ctrl+alt+f1进入命令行模式，登录用户后，关闭显卡驱动

sudo /etc/init.d/lightdm stop

运行cuda安装包，作者全程选yes

sudo ./cuda_8.0.61_375.26_linux.run

安装完成后运行

sudo etc/init.d/gdm  restart

2.安装cudnn

点击下载链接，下载v5.1 forCUDA8.0的linux版本，文件名为cudnn-8.0-linux-x64-v5.1.tgz

cudnn下载链接

解压后进入到文件夹（cuda）中，将其中的文件用以下代码拷贝到cuda的相应目录并更改权限

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

3.安装tensorflow-gpu版本

输入命令

pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.5.0-cp27-none-linux_x86_64.whl

4.运行tensorflow

打开终端输入python，进入后输入

import tensorflow as tf

如果没有报错恭喜你已经成功的安装了linux二进制版本的GPU版本tensorflow并正式加入了TF（tensorflow） Boys/Girls家族，接下来你可以移步tensorflow的中文社区去试试helloworld的小代码并开启你的机器遗忘生涯TensorFlow中文社区-首页

    原文作者：Square Tsou
    原文地址: https://zhuanlan.zhihu.com/p/27700114
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞