Ubuntu18.04 安裝TensorFlow 和 Keras


 

 

 

TensorFlow和Keras是當前兩款主流的深度學習框架,Keras被采納為TensorFlow的高級API,平時做深度學習任務,可以使用Keras作為深度學習框架,並用TensorFlow作為后端引擎。

1、安裝之前,先確認pip包管理器最新:

sudo apt-get update sudo apt-get upgrade sudo apt-get install python-pip python-dev
sudo apt-get install python3-pip python3-dev

使用pip安裝包時,默認安裝的是python2的包。想安裝python3的包,使用pip3。

2、安裝Python相關庫

(1)安裝BLAS庫

 sudo apt-get install build-essential cmake git unzip pkg-config libopenblas-dev liblapack-dev 

(2)安裝python科學套件

 sudo apt-get install python-numpy python-scipy python-matplotlib python-yaml 

 sudo apt-get install python3-numpy python3-scipy python3-matplotlib python3-yaml

(3)安裝HDF5。這個庫最初由NASA開發,用高效的二進制格式來保存數值數據的大文件。(它可以讓你將Keras模型快速高效地保存到磁盤)

 sudo apt-get install libhdf5-serial-dev python-h5py 

sudo apt-get install libhdf5-serial-dev python3-h5py

(4)安裝Graphviz和pydot-ng,這兩個包可以將Keras模型可視化。

 sudo apt-get install graphviz 

 sudo pip install pydot-ng

sudo pip3 install pydot-ng

(5)安裝python-opencv包

 sudo apt-get install python-opencv 

sudo apt-get install python3-opencv

3、安裝CUDA和cuDNN

見另一篇博文:https://www.cnblogs.com/booturbo/p/11834661.html

4、安裝支持GPU的TensorFlow

 pip install tensorflow-gpu 

 pip3 install tensorflow-gpu

卸載Tensorflow-GPU

sudo pip uninstall protobuf sudo pip3 uninstall protobuf pip uninstall protobuf pip3 uninstall protobuf
sudo pip uninstall tensorflow-gpu sudo pip3 uninstall tensorflow-gpu

卸載完成。

5、安裝Keras

可以從PyPI安裝Keras

 sudo pip install keras 

 sudo pip3 install keras

對應的Keras卸載方法:

sudo pip uninstall keras sudo pip3 uninstall keras

然后刪除殘留文件夾

sudo rm -r /usr/local/lib/python2.7/dist-packages/Keras-2.3.1-py2.7.egg sudo rm -r /usr/local/lib/python3.6/dist-packages/Keras-2.3.1-py3.6.egg

這樣便完全刪除了。

也可以從GitHub安裝Keras。這種方式安裝,可以運行keras/examples文件夾里的示例。

git clone https://github.com/fchollet/keras
cd keras sudo python setup.py install

git clone https://github.com/fchollet/keras cd keras sudo python3 setup.py install

安裝完成后,運行下示例腳本,例如 MNIST

在keras文件目錄下運行, python examples/mnist_cnn.py 

 python3 examples/mnist_cnn.py

運行過Keras之后,就可以在~/.keras/keras.json看到Keras的配置文件,編輯該文件為Keras選擇后端引擎,

{
    "image_data_format": "channels_last", "epsilon": le-07,
    "floatx": "float32",
    "backend": "tensorflow" }

 

 安裝TensorFlow2

 參考官方指南 

1、檢查是否配置好Python環境:

python3 --version pip3 --version virtualenv --version

如果已安裝這些軟件包,請跳至下一步。否則,請安裝:

sudo apt update sudo apt install python3-dev python3-pip sudo pip3 install -U virtualenv

2、創建虛擬環境(推薦)

Python虛擬環境用於將軟件包安裝與系統隔離開來。

創建一個新的虛擬環境,方法是選擇Python解釋器並創建一個   ./venv  目錄來存放它:

virtualenv --system-site-packages -p python3 ./venv

使用特定於shell 的命令激活該虛擬環境:

source ./venv/bin/activate

當 virtualenv 處於激活狀態時,shell 提示符帶有(venv)前綴。

在不影響系統設置的情況下,在虛擬環境中安裝軟件包。首先升級pip:

pip install --upgrade pip pip list #展示虛擬環境中安裝的包

以后可以使用以下命令退出 virtualenv

deactivate

3、安裝TensorFlow pip軟件包

在虛擬環境中安裝

pip install --upgrade tensorflow

驗證安裝效果:

python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

安裝成功

 

 

到此結束。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM