1. 程式人生 > >linux 安裝tensorflow(gpu版本)

linux 安裝tensorflow(gpu版本)

pre 源碼編譯安裝 gpu版本 深度學習 ora 報錯 新版 size googl

一、安裝cuda

具體安裝過程見我的另一篇博客,ubuntu16.04下安裝配置深度學習環境

二、安裝tensorflow

1.具體安裝過程官網其實寫的比較詳細,總結一下的話可以分為兩種:安裝release版本和源碼編譯安裝。因為源碼編譯安裝比較繁瑣,且需要安裝谷歌自己的編譯器bazel,所以我選擇安裝編譯好的。

2.我寫這篇博客的時候tensorflow更新到了1.4.0,安裝編譯好的一定看版本,因為每個版本依賴的底層庫是不一樣的。

1.4.0版本安裝之前需要安裝CUDA-8,cuDNN v6.0.,ibcupti-dev library

註意以上軟件的版本,cuda一定要是8,cudnn得是6.0(我之前裝的是1.2.0,則cudnn是v5.1就行,所以這個版本很重要,如果你的cudnn版本過低的話會報錯:ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory)

安裝bcupti-dev library可執行

sudo apt-get install libcupti-dev

3.以上環境準備好後,安裝是很簡單的

如果你用的是anaconda,安裝步驟如下:

conda create -n tensorflow python=2.7 # or python=3.3, etc.
source activate tensorflow
pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.4.0-cp35-cp35m-linux_x86_64.whl

如果是python直接安裝的話:
首先需要pip,如果沒有的話可以使用下面的命令安裝
sudo apt-get install python-pip python-dev   # for Python 2.7
sudo apt-get install python3-pip python3-dev # for Python 3.n

下面使用pip安裝tensorflow

$ pip install tensorflow # Python 2.7; CPU support (no GPU support)

$ pip3 install tensorflow # Python 3.n; CPU support (no GPU support)

$ pip install tensorflow-gpu

# Python 2.7; GPU support

$ pip3 install tensorflow-gpu # Python 3.n; GPU support

不加版本的話默認是最新版本,如果想下特定的版本可在tensorflow後面加上版本號,例如第一個可以寫成pip install tensorflow-1.1.0

三、測試tensorflow是否安裝成功

經過一二兩個步驟,應該是安裝成功了,可寫個tensorflow的小程序測試一下

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

linux 安裝tensorflow(gpu版本)