caffe安裝詳解

一、caffe安裝GPU版本

在安裝GPU版本之前,需要確認一下系統是否支持CUDA。可以通過以下命令查看:

nvcc -V

如果輸出顯示“command not found”,則需要安裝CUDA。具體安裝方法請參考官網教程。

1、從caffe官網(http://caffe.berkeleyvision.org/install_apt.html)下載相應版本的caffe安裝文件

sudo apt-get update
sudo apt-get upgrade

2、下載相關依賴包

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install libatlas-base-dev libopenblas-dev libgflags-dev libgoogle-glog-dev liblmdb-dev

3、解壓caffe安裝文件並編譯

tar -xvf caffe>version<.tar.gz
cd caffe>version<
cp Makefile.config.example Makefile.config
sudo vim Makefile.config

修改Makefile.config文件中相應的配置(比如開啟GPU支持等),保存並退出。

make all -j8
make test -j8
make runtest -j8
sudo make pycaffe -j8

二、conda安裝caffe

1、在anaconda環境下新建虛擬環境。如果你沒有安裝anaconda,可以到官網下載並安裝

conda create --name py27 python=2.7
conda activate py27

2、添加anaconda的安裝目錄到環境變量PATH中

echo export PATH=~/anaconda3/bin:$PATH >> ~/.bashrc
source ~/.bashrc

3、安裝caffe。可以通過所需版本github地址來獲取caffe

conda install caffe

三、caffe安裝筆記本電腦

1、下載需要的依賴包

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install build-essential cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
sudo apt-get install python-dev python-numpy python-py python-pytest python-coverage python-pip libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-dev protobuf-compiler gfortran libjpeg62 libfreeimage-dev libatlas-base-dev git python-dev python-setuptools python-numpy python-scipy libprotobuf-dev libleveldb-dev liblmdb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo pip install -U pip protobuf
sudo pip install --user --install-option="--prefix=" -r python/requirements.txt

2、下載caffe

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

3、添加caffe路徑

echo "export PYTHONPATH=$(pwd)/python:$PYTHONPATH" >> ~/.bashrc
source ~/.bashrc

4、編譯caffe

cd caffe
mkdir build
cd build

打開CPU版本

cmake -DCPU_ONLY=ON ..

或者打開GPU版本

cmake -DUSE_CUDNN=OFF -DUSE_NCCL=OFF ..

編譯

make all -j8
make install
make runtest

四、離線安裝caffe

1、到caffe官網下載相應版本的離線安裝包

2、解壓壓縮包並進入解壓後的目錄

tar -zxvf caffe>version<.tar.gz
cd caffe>version<

3、修改Makefile.config文件,添加相應選項

4、安裝必要依賴項

sudo apt-get update
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install libatlas-base-dev libopenblas-dev libgflags-dev libgoogle-glog-dev liblmdb-dev

5、編譯caffe

make all
make test
make runtest

至此,caffe安裝完畢。

原創文章,作者:UOCL,如若轉載,請註明出處:https://www.506064.com/zh-hant/n/131623.html

(0)
打賞 微信掃一掃 微信掃一掃 支付寶掃一掃 支付寶掃一掃
UOCL的頭像UOCL
上一篇 2024-10-03 23:46
下一篇 2024-10-03 23:46

相關推薦

  • Linux sync詳解

    一、sync概述 sync是Linux中一個非常重要的命令,它可以將文件系統緩存中的內容,強制寫入磁盤中。在執行sync之前,所有的文件系統更新將不會立即寫入磁盤,而是先緩存在內存…

    編程 2025-04-25
  • 神經網絡代碼詳解

    神經網絡作為一種人工智能技術,被廣泛應用於語音識別、圖像識別、自然語言處理等領域。而神經網絡的模型編寫,離不開代碼。本文將從多個方面詳細闡述神經網絡模型編寫的代碼技術。 一、神經網…

    編程 2025-04-25
  • MPU6050工作原理詳解

    一、什麼是MPU6050 MPU6050是一種六軸慣性傳感器,能夠同時測量加速度和角速度。它由三個傳感器組成:一個三軸加速度計和一個三軸陀螺儀。這個組合提供了非常精細的姿態解算,其…

    編程 2025-04-25
  • nginx與apache應用開發詳解

    一、概述 nginx和apache都是常見的web服務器。nginx是一個高性能的反向代理web服務器,將負載均衡和緩存集成在了一起,可以動靜分離。apache是一個可擴展的web…

    編程 2025-04-25
  • 詳解eclipse設置

    一、安裝與基礎設置 1、下載eclipse並進行安裝。 2、打開eclipse,選擇對應的工作空間路徑。 File -> Switch Workspace -> [選擇…

    編程 2025-04-25
  • Python安裝OS庫詳解

    一、OS簡介 OS庫是Python標準庫的一部分,它提供了跨平台的操作系統功能,使得Python可以進行文件操作、進程管理、環境變量讀取等系統級操作。 OS庫中包含了大量的文件和目…

    編程 2025-04-25
  • git config user.name的詳解

    一、為什麼要使用git config user.name? git是一個非常流行的分布式版本控制系統,很多程序員都會用到它。在使用git commit提交代碼時,需要記錄commi…

    編程 2025-04-25
  • Linux修改文件名命令詳解

    在Linux系統中,修改文件名是一個很常見的操作。Linux提供了多種方式來修改文件名,這篇文章將介紹Linux修改文件名的詳細操作。 一、mv命令 mv命令是Linux下的常用命…

    編程 2025-04-25
  • Python輸入輸出詳解

    一、文件讀寫 Python中文件的讀寫操作是必不可少的基本技能之一。讀寫文件分別使用open()函數中的’r’和’w’參數,讀取文件…

    編程 2025-04-25
  • Java BigDecimal 精度詳解

    一、基礎概念 Java BigDecimal 是一個用於高精度計算的類。普通的 double 或 float 類型只能精確表示有限的數字,而對於需要高精度計算的場景,BigDeci…

    編程 2025-04-25

發表回復

登錄後才能評論