Skip to content

Instantly share code, notes, and snippets.

@jetsonhacks

jetsonhacks/installCaffe.sh

Last active July 10, 2020 04:09
Show Gist options
  • Save jetsonhacks/acf63b993b44e1fb9528 to your computer and use it in GitHub Desktop.
Save jetsonhacks/acf63b993b44e1fb9528 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. jetsonhacks revised this gist Jan 18, 2015. 1 changed file with 1 addition and 1 deletion.
    2 changes: 1 addition & 1 deletion installCaffe.sh
    View file
    Original file line number Diff line number Diff line change
    @@ -1,5 +1,5 @@
    #!/bin/sh
    # Install and compile Caffe on Jetson TK1
    # Install and compile Caffe on NVIDIA Jetson TK1 Development Kit
    sudo add-apt-repository universe
    sudo apt-get update
    sudo apt-get install libprotobuf-dev protobuf-compiler gfortran \
  2. jetsonhacks revised this gist Jan 18, 2015. 1 changed file with 0 additions and 5 deletions.
    5 changes: 0 additions & 5 deletions installCaffe.sh
    View file
    Original file line number Diff line number Diff line change
    @@ -8,12 +8,7 @@ libboost-thread-dev libboost-system-dev \
    libatlas-base-dev libhdf5-serial-dev libgflags-dev \
    libgoogle-glog-dev liblmdb-dev -y

    # Make sure the library path is set up to look for cuda 6.5 library
    sudo usermod -a -G video $USER
    echo "# Add CUDA bin & library paths:" >> ~/.bashrc
    echo "export PATH=/usr/local/cuda/bin:$PATH" >> ~/.bashrc
    echo "export LD_LIBRARY_PATH=/usr/local/cuda/lib:$LD_LIBRARY_PATH" >> ~/.bashrc
    source ~/.bashrc

    # Git clone Caffe
    sudo apt-get install -y git
  3. jetsonhacks created this gist Jan 17, 2015.
    28 changes: 28 additions & 0 deletions installCaffe.sh
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,28 @@
    #!/bin/sh
    # Install and compile Caffe on Jetson TK1
    sudo add-apt-repository universe
    sudo apt-get update
    sudo apt-get install libprotobuf-dev protobuf-compiler gfortran \
    libboost-dev cmake libleveldb-dev libsnappy-dev \
    libboost-thread-dev libboost-system-dev \
    libatlas-base-dev libhdf5-serial-dev libgflags-dev \
    libgoogle-glog-dev liblmdb-dev -y

    # Make sure the library path is set up to look for cuda 6.5 library
    sudo usermod -a -G video $USER
    echo "# Add CUDA bin & library paths:" >> ~/.bashrc
    echo "export PATH=/usr/local/cuda/bin:$PATH" >> ~/.bashrc
    echo "export LD_LIBRARY_PATH=/usr/local/cuda/lib:$LD_LIBRARY_PATH" >> ~/.bashrc
    source ~/.bashrc

    # Git clone Caffe
    sudo apt-get install -y git
    git clone https://github.com/BVLC/caffe.git
    cd caffe && git checkout dev
    cp Makefile.config.example Makefile.config

    make -j 4 all

    make -j 4 runtest

    build/tools/caffe time --model=models/bvlc_alexnet/deploy.prototxt --gpu=0