Ubuntu 18.04 源碼安裝 Caffe 步驟

Caffe 是一個深度學習框架，特別適合於圖像分類和卷積神經網絡（CNN）的研究。它由伯克利人工智慧研究中心（BAIR）開發，並以其速度和模組化設計而聞名。本文將介紹如何在 Ubuntu 18.04 上從源碼安裝 Caffe，幫助您快速上手這個強大的工具。

前置條件

在開始安裝之前，您需要確保系統中已安裝以下依賴項：

• Ubuntu 18.04
• git
• cmake
• g++
• protobuf
• OpenCV
• BLAS（如 ATLAS 或 OpenBLAS）
• Python 2.7 或 3.x

安裝依賴項

首先，您需要更新系統並安裝必要的依賴項。打開終端並執行以下命令：

sudo apt update
sudo apt install -y build-essential cmake git libatlas-base-dev libprotobuf-dev protobuf-compiler libopencv-dev python3-dev python3-numpy

下載 Caffe 源碼

接下來，您需要從 GitHub 上下載 Caffe 的源碼。執行以下命令：

git clone https://github.com/BVLC/caffe.git
cd caffe

配置 Caffe

在下載完成後，您需要配置 Caffe。首先，複製示例配置文件：

cp Makefile.config.example Makefile.config

然後，使用文本編輯器打開 Makefile.config 文件，根據您的需求進行修改。特別是，您可以選擇啟用 GPU 支持或設置 Python 路徑。

啟用 GPU 支持（可選）

如果您的系統支持 GPU，您可以在 Makefile.config 中取消註釋以下行：

# USE_CUDA := 1

編譯 Caffe

配置完成後，您可以開始編譯 Caffe。執行以下命令：

make all
make test
make runtest

這些命令將編譯 Caffe 的核心庫、測試和運行測試。

安裝 Python 接口

如果您需要使用 Python 接口，請執行以下命令：

make pycaffe

確保將 Caffe 的 Python 路徑添加到您的環境變量中。您可以在 ~/.bashrc 文件中添加以下行：

export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH

驗證安裝

安裝完成後，您可以通過在 Python 中導入 Caffe 來驗證安裝是否成功：

python3 -c "import caffe"

如果沒有錯誤消息，則表示 Caffe 安裝成功。

總結

在本文中，我們介紹了如何在 Ubuntu 18.04 上從源碼安裝 Caffe。這些步驟涵蓋了從安裝依賴項到編譯和驗證安裝的整個過程。Caffe 是一個強大的深度學習框架，適合各種應用場景。如果您需要穩定的環境來運行 Caffe，考慮使用 香港VPS 服務，以獲得更好的性能和靈活性。