1. 程式人生 > >ubuntu16.04 安裝tensorflow(GPU、CPU)

ubuntu16.04 安裝tensorflow(GPU、CPU)

連結:

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

【注:需要更改成清華源】

GPU安裝

適用GTX1050,GTX1060,GTX1070,GTX1080等高階顯示卡系列

  1. 安裝好系統後,更新驅動。系統設定-》軟體和更新-》附加驅動-》更新nvidia和intel
  2. 安裝cuda9

下載 cuda9.run,(deb格式容易出錯誤)。執行如下命令:

sudo sh cuda_9.0.176_384.81_linux.run

accept

n(不要安裝driver)

y

y

y

安裝完成後,設定環境變數。

設定環境變數:

ctrl+H 開啟 (主目錄/home/*/) .bashrc檔案,新增

export PATH=/usr/local/cuda-9.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH  

##############注意 解決import tensorflow 錯誤的問題############## 

更改為:(這裡執行,不會再出現錯誤。很抱歉。)

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0

終端執行:source ~/.bashrc

檢查:nvcc --version

3. 安裝cudnn7

下載 for linux版本的,執行如下命令:

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

sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

4. 安裝anaconda(python3.5)

bash Anaconda3-4.2.0-Linux-x86_64.sh

yes(設定環境變數)

5. 安裝tensorflow-gpu(任意版本)

複製開始的藍色連結上的命令。


CPU安裝:

直接複製開始連結上的命令安裝tensorflow-cpu。