Ubuntu16.04 pip下载安装tensorflow(GPU版)
生活随笔
收集整理的這篇文章主要介紹了
Ubuntu16.04 pip下载安装tensorflow(GPU版)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
通過編譯tensorflow編譯源碼再安裝的方法(如本人博客Ubuntu16.04 源碼安裝tensorflow(GPU版) 中所講),安裝過的朋友就知道,這是個非常浪費時間的過程,最終編譯生成一個whl文件,通過pip安裝這個whl文件就行了。其實pip人家鏡像源里面有一些常用版本的whl文件,不用咱們編譯,直接下下來安裝就可以了。
下面就介紹直接用pip下載安裝tensorflow的方法。
cuda及cuDNN安裝
這部分參見本人另外一篇博客Ubuntu16.04 Caffe安裝筆記 中的一二三章節。
現在默認安裝的tensorflow是1.4版本,要求安裝cuda8.0和cudnn6
安裝依賴
sudo apt-get install libcupti-devsudo apt-get install python-pip python-dev # for Python 2.7 sudo apt-get install python3-pip python3-dev # for Python 3pip install --upgrade pip pip3 install --upgrade pip安裝tensorflow-gpu版本
根據自己用python2.7還是Python3來選擇裝相應的版本(==后面為需要安裝的版本號)
pip install tensorflow-gpu==1.8.0 # Python 2.7 pip3 install tensorflow-gpu==1.8.0 # Python 3測試
因為我安裝的是python3版本
總結
以上是生活随笔為你收集整理的Ubuntu16.04 pip下载安装tensorflow(GPU版)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: C#托管代码与C++非托管代码互相调用一
- 下一篇: adb logcat 基本用法