Ubuntu16.04+cuda8.0+cudnn6.0+tensorflow（四）

来源：互联网发布：我的世界0.15.1枪械js 编辑：程序博客网时间：2024/06/08 06:46

边走边说我踩过的那些坑儿

Ubuntu16.04+CUDA8.0+cudnn6.0下安装tensorflow

参考链接1：https://www.tensorflow.org/install/install_linux

准备工具：

1.显卡和显卡驱动

2. Cuda toolkit

3. cudnn 6.0

4. Anaconda （python 2.7）

步骤：

########### 准备工作 ######

#### 装显卡和安装显卡驱动

#### 安装CUDA toolkit

#### 安装cudnn 6.0

#########################

第一步：安装libcupti-dev

sudo apt-get install libcupti-dev

第二步：安装Anaconda

先下载conda安装包，可到清华镜像下载：

https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/

打开终端，切换到conda安装包的路径，输入

bash Anaconda2-5.0.0.1-Linux-x86_64.s

第三步：安装tensorflow

1. 创建conda环境：

$ conda create -n tensorflow

2. 激活conda 环境：

$ source activate tensorflow (tensorflow)$  # Your prompt should change

3. 安装conda：

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.3.0-cp27-none-linux_x86_64.whl

这里选择的是gpu版本

第四步：测试gpu版本的tensorflow安装是否成功

打开终端，输入Python，接着输入以下命令：

import tensorflow as tfhello = tf.constant('Hello, TensorFlow!')sess = tf.Session()print(sess.run(hello))

正常输出结果，则测试成功

阅读全文

0 0