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

Ubuntu 16.04 + GTX 1080 + CUDA 8.0 + OpenCV 3.1.0 + cuDNN-v5 + Caffe 安裝教程

Ubuntu 16.04 + GTX 1080 + CUDA 8.0 + OpenCV 3.1.0 + cuDNN-v5 + Caffe 安裝教程

在深度學習和計算機視覺的領域中，搭建一個高效的開發環境是至關重要的。本文將介紹如何在 Ubuntu 16.04 系統上安裝 NVIDIA GTX 1080 顯示卡的驅動程式、CUDA 8.0、cuDNN v5、OpenCV 3.1.0 以及 Caffe 框架。這些工具的組合能夠為開發者提供強大的計算能力和靈活的圖像處理功能。

前期準備

  • 確保你的系統是 Ubuntu 16.04。
  • 擁有 NVIDIA GTX 1080 顯示卡。
  • 擁有 sudo 權限的用戶。

安裝 NVIDIA 驅動程式

首先，更新系統並安裝必要的工具：

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install build-essential dkms

接下來，添加 NVIDIA 的 PPA 並安裝驅動程式：

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-384

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

sudo reboot

安裝 CUDA 8.0

前往 NVIDIA 的官方網站下載 CUDA 8.0 的安裝包。選擇適合 Ubuntu 16.04 的版本，然後使用以下命令安裝：

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.XX-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu1604-8-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

安裝完成後，將 CUDA 的路徑添加到環境變量中：

echo 'export PATH=/usr/local/cuda-8.0/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc

安裝 cuDNN v5

下載 cuDNN v5 的壓縮包，並解壓縮後將其內容複製到 CUDA 的相應目錄中：

tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

安裝 OpenCV 3.1.0

首先，安裝 OpenCV 的依賴項：

sudo apt-get install libopencv-dev python-opencv

接下來，從 GitHub 下載 OpenCV 3.1.0 的源碼：

git clone https://github.com/opencv/opencv.git
cd opencv
git checkout 3.1.0
mkdir build
cd build
cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..
make -j4
sudo make install

安裝 Caffe 框架

首先，安裝 Caffe 的依賴項：

sudo apt-get install libatlas-base-dev
sudo apt-get install protobuf-compiler
sudo apt-get install libprotobuf-dev
sudo apt-get install libhdf5-serial-dev
sudo apt-get install libleveldb-dev
sudo apt-get install libsnappy-dev
sudo apt-get install libopencv-dev
sudo apt-get install libboost-all-dev

然後，從 GitHub 下載 Caffe 的源碼並編譯：

git clone https://github.com/BVLC/caffe.git
cd caffe
cp Makefile.config.example Makefile.config
# 編輯 Makefile.config，根據需要啟用 GPU 支持
make all -j8
make test -j8
make runtest

總結

通過以上步驟，我們成功在 Ubuntu 16.04 系統上安裝了 NVIDIA GTX 1080 驅動程式、CUDA 8.0、cuDNN v5、OpenCV 3.1.0 和 Caffe 框架。這些工具的組合將為開發者提供強大的計算能力和靈活的圖像處理功能，適合進行深度學習和計算機視覺的研究與開發。

如果您需要穩定的 VPS 服務來支持您的開發工作，請考慮我們的解決方案，提供高效能的計算環境，助您在技術上更上一層樓。

You may also like...