Ubuntu 18.04 安裝 Caffe 具體步驟

Caffe 是一個流行的深度學習框架，特別適合於圖像分類和卷積神經網絡（CNN）的研究。本文將詳細介紹在 Ubuntu 18.04 上安裝 Caffe 的具體步驟，幫助讀者快速上手。

前期準備

在開始安裝之前，請確保您的系統已經更新到最新版本。可以使用以下命令進行更新：

sudo apt update
sudo apt upgrade

接下來，您需要安裝一些必要的依賴項，包括編譯工具和庫。執行以下命令：

sudo apt install build-essential cmake git libatlas-base-dev libboost-all-dev libhdf5-dev protobuf-compiler libprotobuf-dev libopencv-dev libgflags-dev libgoogle-glog-dev libgtest-dev liblmdb-dev libsnappy-dev libopenblas-dev

安裝 Caffe

1. 下載 Caffe 源碼

使用 Git 下載 Caffe 的源碼：

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

2. 配置 Caffe

在 Caffe 目錄中，您需要複製一個示例配置文件並進行編輯：

cp Makefile.config.example Makefile.config
nano Makefile.config

在編輯器中，您可以根據需要啟用或禁用某些功能。確保將以下行取消註釋以啟用 GPU 支持（如果您的系統支持 NVIDIA GPU）：

USE_CUDA := 1
CUDA_DIR := /usr/local/cuda

3. 編譯 Caffe

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

make -j$(nproc)

這將使用所有可用的處理器核心來加速編譯過程。編譯完成後，您可以運行以下命令來測試 Caffe 是否正確安裝：

make test

4. 安裝 Python 接口（可選）

如果您希望使用 Python 接口，則需要安裝 Python 相關的依賴項。首先，安裝 Python 和 pip：

sudo apt install python3-dev python3-pip

然後，安裝 Caffe 的 Python 接口：

cd python
for req in $(cat requirements.txt); do pip3 install $req; done

最後，將 Caffe 的 Python 路徑添加到您的環境變量中。編輯您的 .bashrc 文件：

nano ~/.bashrc

在文件末尾添加以下行：

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

請將 “/path/to/caffe” 替換為您實際的 Caffe 安裝路徑。保存並退出編輯器，然後運行以下命令使更改生效：

source ~/.bashrc

測試 Caffe 安裝

要測試 Caffe 是否正確安裝，您可以運行以下 Python 代碼：

import caffe
print(caffe.__version__)

如果沒有錯誤，並且顯示了版本號，則表示 Caffe 安裝成功。

總結

本文介紹了在 Ubuntu 18.04 上安裝 Caffe 的具體步驟，包括必要的依賴項、編譯過程以及 Python 接口的安裝。透過這些步驟，您可以輕鬆地在自己的環境中設置 Caffe，開始進行深度學習的實驗和開發。

如果您需要穩定的 VPS 來運行您的深度學習項目，Server.HK 提供多種選擇，適合不同需求的用戶。無論是 香港VPS 還是其他類型的 伺服器，都能滿足您的需求。