Google Colab - 安裝機器學習庫



Colab 支援市場上大多數機器學習庫。在本章中,讓我們快速瞭解如何在你的 Colab 筆記本中安裝這些庫。

要安裝庫,你可以使用以下任一選項:

!pip install

!apt-get install

Keras

Keras 使用 Python 編寫,執行在 TensorFlow、CNTK 或 Theano 之上。它可以輕鬆快速地構建神經網路應用程式原型。它支援卷積網路 (CNN) 和迴圈網路,以及它們的組合。它無縫支援 GPU。

要安裝 Keras,請使用以下命令:

!pip install -q keras

PyTorch

PyTorch 非常適合開發深度學習應用程式。它是一個最佳化的張量庫,並且支援 GPU。要安裝 PyTorch,請使用以下命令:

!pip3 install torch torchvision

MxNet

Apache MxNet 是另一個靈活高效的深度學習庫。要安裝 MxNet,請執行以下命令:

!apt install libnvrtc8.0
!pip install mxnet-cu80

OpenCV

OpenCV 是一個開源計算機視覺庫,用於開發機器學習應用程式。它擁有超過 2500 個最佳化的演算法,支援多種應用,例如人臉識別、物體識別、運動物體跟蹤、影像拼接等等。像 Google、Yahoo、Microsoft、Intel、IBM、Sony、Honda、Toyota 等巨頭都使用這個庫。它非常適合開發即時視覺應用程式。

要安裝 OpenCV,請使用以下命令:

!apt-get -qq install -y libsm6 libxext6 && pip install -q -U opencv-python

XGBoost

XGBoost 是一個分散式梯度提升庫,可以在 Hadoop 等主要分散式環境中執行。它高效、靈活且可移植。它在梯度提升框架下實現了機器學習演算法。

要安裝 XGBoost,請使用以下命令:

!pip install -q xgboost==0.4a30

GraphViz

Graphviz 是一個用於圖形視覺化的開源軟體。它用於網路、生物資訊學、資料庫設計以及許多需要資料視覺化介面的領域的視覺化。

要安裝 GraphViz,請使用以下命令:

!apt-get -qq install -y graphviz && pip install -q pydot

到目前為止,你已經學會了如何建立包含流行機器學習庫的 Jupyter 筆記本。你現在可以開始開發你的機器學習模型了。這需要強大的處理能力。Colab 為你的筆記本提供免費的 GPU。

在下一章中,我們將學習如何為你的筆記本啟用 GPU。

廣告

© . All rights reserved.