1. 程式人生 > >linux下安裝tensorflow簡潔教程

linux下安裝tensorflow簡潔教程

 

  1. 安裝CUDAcuDNN()

    cat /usr/local/cuda/version.txt檢視CUDA版本

    需要將cuda加入到環境變數

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

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

    然後啟用source ~/.bashrc

    這個時候nvcc –V就可以檢視CUDA版本了。

 

2.安裝anaconda

 

新建資料夾用於放置安裝包 mkdir packages

 

 

wget –c https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.2.0-Linux-x86_64.sh

 

注:下載連結(到清華源找到5.2版本點選右鍵複製連結地址即可)安裝5.2版本。不用建立虛擬環境直接安裝tensorflow

 

 sh Anaconda3-5.3.0-Linux-x86_64.sh      安裝檔案

 

3.安裝tensorflow

 

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ tensorflow-gpu

 

清華源

 

注:如果沒有自動安裝對應的3.6版本。

 

則先建立tensorflow環境conda create –n tensorflow python=3.6

 

   啟用               conda activate tensorflow

 

再安裝即可.

 

測試是否安裝成功。

 

python 

 

 import tensorflow  (以後每次都需要先進去tensorflow環境)