一份Linux下建立tensorflow環境的教程想分享給有需要的人~
安裝步驟
1. update下載下傳最新的軟體安裝包,upgrade安裝已下載下傳的最新的安裝包
sudo apt update
sudo apt upgrade

2. 添加cuda環境變量
進入個人目錄下,打開 .bashrc 檔案,在後面添加cuda路徑
export PATH=/usr/local/cuda-8.0/bin/:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
儲存退出後, 運作 source .bashrc ,并通過 nvcc -V指令檢視是否添加成功。
3. 安裝Anaconda
在某個使用者目錄下下載下傳好Anaconda 的安裝檔案,可以直接複制相應版本的安裝檔案到自己的使用者目錄下進行安裝。
運作 bash Anaconda2-5.0.1-Linux-x86_64.sh 進行安裝
此處選擇yes,自動将Anaconda2添加到 .bashrc 中的PATH變量中。
運作source.bashrc 然後輸入Python,現在Python運作的已經是Anconda的Python
現在Anaconda已經安裝好了。因為Anaconda.org的伺服器在國外,下載下傳速度比較慢,可以使用清華的鏡像:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
4. 建立Anaconda虛拟環境
conda create -n 虛拟環境名字 python=版本号
可以通過 conda info -e 指令檢視目前所有的Anaconda虛拟環境。預設目前所在的環境是root環境,通過 source activate 虛拟環境名 來激活對應的虛拟環境。
5. 安裝TensorFlow-GPU
可以選擇将tensorflow-gpu安裝到指定的虛拟環境中,此處安裝到剛剛建立的 tf13中。通過Anaconda可以安裝指定版本的tensorflow,此處安裝tensorflow1.3 版本。TensorFlow已經更新到1.4版本,但是通過Anaconda源中最新的tensorflow隻有1.3版本的,想要安裝tensorflow1.4版本的可以安裝官網上教程進行安裝。
驗證tensorflow是否安裝成功
6. 安裝keras-gpu
驗證keras是否安裝成功。
至此,Anaconda+ tensorflow + keras 已經安裝成功了。