边走边说我踩过的那些坑儿
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这里选择的是gpu版本https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.3.0-cp27-none-linux_x86_64.whl
第四步:测试gpu版本的tensorflow安装是否成功
打开终端,输入Python,接着输入以下命令:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
正常输出结果,则测试成功