服务器设置和教程 · 8 10 月, 2024

Ubuntu 16.04 + CUDA 7.5 + Caffe 深度學習環境搭建

Ubuntu 16.04 + CUDA 7.5 + Caffe 深度學習環境搭建

隨著深度學習技術的迅速發展，許多研究者和開發者都希望能夠搭建一個高效的深度學習環境。本文將介紹如何在 Ubuntu 16.04 上安裝 CUDA 7.5 和 Caffe，幫助您快速建立深度學習的開發環境。

環境準備

在開始之前，您需要確保您的系統符合以下要求：

  • 操作系統：Ubuntu 16.04
  • 顯示卡：NVIDIA 顯示卡，支持 CUDA
  • 網絡連接：需要下載相關的安裝包

安裝 NVIDIA 驅動

首先，您需要安裝 NVIDIA 驅動。可以通過以下命令來安裝：

sudo apt-get update
sudo apt-get install nvidia-352

安裝完成後，重啟系統以使驅動生效。

安裝 CUDA 7.5

接下來，您需要安裝 CUDA 7.5。可以從 NVIDIA 的官方網站下載 CUDA 7.5 的安裝包。下載完成後，使用以下命令進行安裝：

sudo sh cuda_7.5_352.39_linux.run

在安裝過程中，您可以選擇安裝驅動程序或不安裝，因為您已經在前面安裝了驅動。安裝完成後，您需要將 CUDA 的路徑添加到環境變量中。編輯 ~/.bashrc 文件，添加以下行：

export PATH=/usr/local/cuda-7.5/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

然後執行以下命令使改動生效：

source ~/.bashrc

安裝 Caffe

現在，我們可以開始安裝 Caffe。首先，您需要安裝一些依賴包：

sudo apt-get install build-essential cmake git libatlas-base-dev libboost-all-dev libhdf5-serial-dev protobuf-compiler python-protobuf python-numpy python-scipy libopencv-dev

接下來，從 GitHub 上克隆 Caffe 的代碼庫：

git clone https://github.com/BVLC/caffe.git
cd caffe
git checkout tags/r0.15

然後，您需要編輯 Caffe 的配置文件，通常是位於 caffe 目錄下的 Makefile.config.example。將其複製為 Makefile.config 並根據需要進行修改：

cp Makefile.config.example Makefile.config

在 Makefile.config 中，確保以下行被取消註釋以啟用 CUDA 支持：

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

完成後，您可以編譯 Caffe：

make all -j8
make test -j8
make runtest -j8

如果一切順利，您應該能夠成功編譯 Caffe。

驗證安裝

最後，您可以通過運行以下命令來驗證 Caffe 是否安裝成功：

cd python
for req in $(cat requirements.txt); do pip install $req; done
python -c "import caffe; print(caffe.__version__)"

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

總結

本文介紹了如何在 Ubuntu 16.04 上搭建 CUDA 7.5 和 Caffe 的深度學習環境。這一過程涉及到 NVIDIA 驅動的安裝、CUDA 的配置以及 Caffe 的編譯。透過這些步驟，您可以為深度學習項目提供一個穩定的基礎。

如果您需要更高效的計算資源，可以考慮使用 香港VPS 服務，這將為您的深度學習項目提供更好的支持。

You may also like...